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

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

1541번: 잃어버린 괄호

일단 덧셈을 먼저 다 해준 다음에 빼주면 되니까 '-'를 기준으로 split() 해준다. 리스트의 첫 원소에 숫자만 있을 수도, + 연산자가 같이 있을 수도 있으므로 일단 '+'를 기준으로 split() 한 뒤 더해준 결과를 answer에 저장한다. 그 뒤에 똑같이 '+'를 기준으로 split() 한 뒤 더해준 결과를 answer에서 계속 빼주면 된다.

 

처음에는 eval() 함수를 이용해서 풀려고 했는데.. 0으로 시작하는 숫자도 있어서 당연히 런타임 에러가 났다. 문제를 똑바로 읽자..... 

import sys
input = sys.stdin.readline
formula = input().split('-')
answer = sum(map(int, formula[0].split('+')))
for i in range(1, len(formula)):
    answer -= sum(map(int, formula[i].split('+')))
print(answer)