백준 1541: 잃어버린 괄호 (Python)

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

1541번: 잃어버린 괄호

 

s = input()
s = s.split('-')
s2 = []
res = 0

if s[0].isdigit():
    res += int(s[0])
else:
    s2 = s[0].split('+')
    for i in s2:
        res += int(i)
del s[0]

for i in s:
    if i.isdigit():
        res -= int(i)
    else:
        s2.clear()
        s2 = list(map(int, i.split('+')))
        res -= sum(s2)

print(res)

 

 

맞긴 맞았는데,,, 코드가 어마무시하게 길어서,, 다시 다듬어봤다,,

 

s = input().split('-')
res = 0

for i in range(len(s)):
    for j in s[i].split('+'):
        if i == 0:
            res += int(j)
        else:
            res -= int(j)

print(res)

 

다시 생각해보니 굳이 숫자냐고 묻지 않아도 됐음,,,

 

첫 코드는 잘못 짜서 버렸는데 왜 시간이 제일 짧은건지,,

 

나의 시행착오 ㅠㅠ