알고리즘 도전기 - 36

1541 잃어버린 괄호

문제

세준이는 양수와 +, -, 그리고 괄호를 가지고 식을 만들었다. 그리고 나서 세준이는 괄호를 모두 지웠다.
그리고 나서 세준이는 괄호를 적절히 쳐서 이 식의 값을 최소로 만들려고 한다.
괄호를 적절히 쳐서 이 식의 값을 최소로 만드는 프로그램을 작성하시오.

입력

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

출력

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

예제 입력

55-50+40

예제 출력

-35

코드

import sys

form = sys.stdin.readline().rstrip()
arr = []
cur = ''
mid_sum = 0
for i in range(len(form)):
    if i == len(form) - 1:
        cur += form[i]
        arr.append(int(cur))
        cur = ''

    if form[i] == '-' or form[i] == '+':
        arr.append(int(cur))
        arr.append(form[i])
        cur = ''
    else:
        cur += form[i]

result = arr[0]

for j in range(len(arr) - 2, 0, -1):
    if arr[j] == '+' and j == 1:
        mid_sum += arr[j + 1]
        result += mid_sum
        break

    if arr[j] == '-':
        mid_sum += arr[j + 1]
        result -= mid_sum
        mid_sum = 0
    elif arr[j] == '+':
        mid_sum += arr[j + 1]

print(result)