코테 준비/문자열, 내장함수

[백준] 1541. 잃어버린 괄호

imsmile2000 2023. 1. 8. 00:40

'-'가 나오면 이후의 숫자들은 괄호 처리에 의해 +가 모두 -로 변한다는 점에 착안

import sys
import re
expression=sys.stdin.readline().rstrip()
num=re.split('[-+]',expression)
oper=re.findall('[-+]',expression)
for i in range(len(oper)-1):
    if oper[i]=='-':
        oper[i+1:]='-'  #-가 있으면 그 이후의 숫자들은 다 빼줌 괄호쳐주는 것과 같은 원리
result=int(num[0])
for i in range(1,len(num)):
    if oper[i-1]=='+':
        result+=int(num[i])
    else:
        result-=int(num[i])
print(result)