BOJ 1541번 : 잃어버린 괄호 (Python/Silver 2)

1541번: 잃어버린 괄호

문제

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

 

입력

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

 

출력

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

---

예제 입력

55-50+40

예제 출력

-35

---

문제 설명

주어진 식에 괄호를 쳐서 결과가 최소가 되게 만들면 된다.

 

풀이

만일 입력이 10-20+30-40이라고 가정하자. 

10-20+30-40 = - 20

---

+를 최대로 만들고 앞에 -를 붙이면 최솟값을 만들 수 있다.

10 - (20 + 30 ) - 40 = - 80

---

따라서 먼저 '-'를 기준으로 문자열을 구분한다.  

split()을 써서 num에 저장한 결과이다.

---

첫 번째 피연산자는 -이든 +이든 무조건 값에 포함되므로 첫 번째 수는 먼저 따로 더한다. 두 번째 피연산자부터는 '+'로 구분하여 그 수들을 빼주면 된다.

 

 

주의할 점: split('-')를 사용했어도 '-'를 기준으로 리스트에 저장된 것이 아닌 '1', '0', '2' - - - 이렇게 저장되어 있다. split(' ')을 사용한 이유가 이와 같다.

Solution

import sys
input = sys.stdin.readline

num = input().split('-')
ans = 0

for i in num[0].split('+'): 
    ans += int(i)
for i in num[1:]:
    for j in i.split('+'):
        ans -= int(j)

print(ans)