백준 1541번: 잃어버린 괄호 (파이썬)

https://www.acmicpc.net/problem/1541

1541번: 잃어버린 괄호

 

문제

 

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

 

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

 

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

 

입력

 

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

 

출력

 

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

 

풀이

 

더하기와 빼기만 있는 연산에서 괄호를 사용해 식을 최소로 만들기 위해서는, 빼는 수를 최대한 크게 만들면 된다.

즉 모든 연속된 더하기에만 괄호를 쳐서 수를 최대한 크게 만든 뒤, 마지막에 모두 빼면 된다.

 

이를 위해서 우선 식을 마이너스 기호를 기준으로 자른다.

이렇게 자르면 첫 번째 덩어리에서 나머지 덩어리들을 모두 뺀 값이 식의 최솟값이다.

예를 들어 20-50+30+70-10+30이라고 하면 ['20', '50+30+70', '10+30']와 같이 잘리게 될 것이고, 20 - 150 - 40 = -170이 이 식의 최솟값이다.

각 덩어리들은 문자열이므로, 문자열을 식으로 변환한 후 계산하여 정수로 배열에 다시 저장한다.

이후 첫 번째 덩어리에서 나머지 덩어리들을 모두 뺀다.

 

소스 코드

 

tokens = input().split('-')

for i in range(len(tokens)):
    token = tokens[i].split('+')
    sum = 0
    for j in range(len(token)):
        sum += int(token[j])
    tokens[i] = sum

result = tokens[0]
for i in range(1, len(tokens)):
    result -= tokens[i]
print(result)