[BOJ 1541, Python 3] 잃어버린 괄호

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

1541번: 잃어버린 괄호

 

 

풀이

처음 - 부호가 나오기 전까지는 더해주다가 -가 나온 이후에는 계속 빼주면 됩니다.

 

왜냐하면 ㅁ + ㅁ - ㅁ + ㅁ + ㅁ + ㅁ + ㅁ 이렇게 마이너스가 하나만 있는 식이 있다고 하면 

ㅁ + ㅁ - (ㅁ + ㅁ + ㅁ - ㅁ + ㅁ) 이렇게 괄호를 해주면 되고

ㅁ + ㅁ - ㅁ + ㅁ - ㅁ + ㅁ - ㅁ 같이 마이너스가 여러 개 있는 식이 있다고 하면

ㅁ + ㅁ - (ㅁ + ㅁ) - (ㅁ + ㅁ) - ㅁ 같이 마이너스와 마이너스 부호 사이를 괄호로 감싸주면 됩니다.

 

코드

def pm():
    global ans
    if flag: ans -= int(val)
    else: ans += int(val)
 
s = input()
flag = False
ans = 0
val = ''
for i in range(len(s)):
    if s[i] in ['+', '-']:
        pm()
        val = ''
        if s[i] == '-': flag = True
    else: val += s[i]
else: pm()
print(ans)