잃어버린 괄호 - 1541번

문제

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

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

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

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

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

풀이

문제에서 "괄호 넣기", "연산자 끼워넣기" 이런 것만 보면 뇌가 얼어버리는 악습관이 있다... 트라우마인가?
그냥 어렵게 생각할 것 없이 처음 인풋을 스트링으로 접근하면 된다는 생각을 하면 되는데!

풀이의 핵심은 마이너스 연산자 직전에 괄호를 넣으면 된다는 점에서 출발한다. 괄호가 없는 식이 주어졌을 때 '-' 이후에 '+'가 나오면 '-' 뒤에 괄호를 쳤을 때 뒤에 값들의 합이 음수로 변하기 때문에 값이 작아지게 된다. 결국 '-'를 기준으로 해서 괄호를 치면 된다.

다음의 코드는 정답이다.

import sys
input = sys.stdin.readline
line = input().split('-')

num = []
for i in line:
    tmpsum = 0
    tmp = i.split('+')
    for j in tmp:
        tmpsum += int(j)
    num.append(tmpsum)

n = num[0]

for i in num[1:]:
    n -= i

print(n)