[Baekjoon] 1541번 잃어버린 괄호

문제

세준이는 양수와 +, -, 그리고 괄호를 가지고 식을 만들었다. 그리고 나서 세준이는 괄호를 모두 지웠다.

그리고 나서 세준이는 괄호를 적절히 쳐서 이 식의 값을 최소로 만들려고 한다.

괄호를 적절히 쳐서 이 식의 값을 최소로 만드는 프로그램을 작성하시오.

입력

첫째 줄에 식이 주어진다. 식은 ‘0’~‘9’, ‘+’, 그리고 ‘-’만으로 이루어져 있고, 가장 처음과 마지막 문자는 숫자이다. 그리고 연속해서 두 개 이상의 연산자가 나타나지 않고, 5자리보다 많이 연속되는 숫자는 없다. 수는 0으로 시작할 수 있다. 입력으로 주어지는 식의 길이는 50보다 작거나 같다.

출력

첫째 줄에 정답을 출력한다.

예제

Example 1:

Input: 
55-50+40
Output: 
-35

조건

시간 제한 : 2초

메모리 제한 : 128 MB

풀이과정

풀이

import sys
inp = (sys.stdin.readline())
li = inp.split('-')
ans = 0
count = 0
for i in li:
  sub = 0
  li2 = i.split('+')
  if not li2[0]:
    sub = 0
  else:
    for j in li2:
      sub += int(j)
  if count == 0:
    ans += sub
    count = 1
  else:
    ans -= sub
print(ans)

그리디와 관련된 문제는 풀이방법에 왜 그리디인가 정당성을 부여해주어야 한다. 따라서 다음과 같이 생각했다.

1. 이 문제가 원하는것은? 괄호를 적절히 쳐서 식의 값을 최소로 만들어야 한다.
2. 마이너스가 왔을 때 다음 마이너스가 올 때까지 전부 괄호를 치면 최소가 된다.
3. 따라서 마이너스로 split을 하자!

위 생각에 따라 마이너스로 split을 하였고 위 생각에 맞추어 코드를 짰다. count의 경우 -로 나누었다 하더라도 맨 처음수는 그렇지 않기 때문이다. 아직 생각을 코드로 구현하는 부분이 부족한 것같다. 이는 많은 문제를 풀면서 해결해야 할 것이다.