파이썬) 백준 알고리즘 | 1541번 : 잃어버린 괄호

문제

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

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

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

입력

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

출력

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

백준 1541 '잃어버린 괄호' 문제입니다.

-를 기준으로 괄호를 치면 최솟값이 됩니다.

그래서 -를 기준으로 split함수로 나누고 그 안에서 +기호가 있으면 더해주면 됩니다.

예를 들어 50-40+30+30-20+30+30-20 이렇게 인풋이 들어왔을 때

50-(40+30+30)-(20+30+30)-20으로 나눌 수 있습니다. -를 기준으로만 split하면 되는 문제입니다.

import sys

string = sys.stdin.readline().strip()
string = string.split('-')
result = int(sum([int(x) for x in string[0].split('+')]))
for st in string[1:]:
    st = st.split('+')
    result -= sum([int(x) for x in st])
print(result)