[백준] 1541 잃어버린 괄호 (python)

문제

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

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

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

입력

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

출력

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

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

1541번: 잃어버린 괄호

---

문제 파악

괄호를 쳐서 식의 값을 최솟값으로 만들려면

전체 식에 대하여 빼기 부호가 나온 뒤의 항에서 괄호를 시작하고 그 다음 빼기 부호가 나오기 직전의 항에서 괄호를 닫는 과정을 진행하면 된다.

 

입력된 식(문자열)을

'-'를 기준으로 split해준 결과로서의 리스트를 a라고 하자.

 

a[0]은 첫번째 '-'가 나오기 전까지의 덩어리이다. 

이 덩어리에는 하나의 항이 있을 수도 있지만 둘 이상의 항이 존재할 수도 있다.

따라서, 이 덩어리를 '+'를 기준으로 split한다.

split한 결과 나뉘어진 항들이 존재한다면 그 값들은 for문을 통해 합해져 ans에 저장된다.

 

첫 번째 덩어리를 마무리하고 이제 나머지(두 번째 이하) 덩어리를 처리할 차례이다. 이 나머지 덩어리들은 모두 '-'를 기준으로 묶여있는 상태이므로 각 항들을 모두 합한 값을 ans에서 빼주면 된다.

 

코드는 아래와 같다. 

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

for i in a[0].split('+'):
    ans += int(i)

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