[백준 1541] 알고리즘 16일차 : 잃어버린 괄호 (python)

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

1541번: 잃어버린 괄호

[그리디 알고리즘] Python

문자열 처리를 쉽게 하고자 파이썬으로 풀어봤다

(백준 문제로는 거의 처음 파이썬으로 푼 듯)

 

접근방법은 식에서 -인 부분만 빼고 괄호로 묶으면 됨

50-(45+90)-(30+40) 이런식으로

 

split 사용해서 -기호 기준으로 분할하고 그 내부는 모두 덧셈일테니까

더하기 계산하고 하나씩 빼가면 된다.

 

근데 맨 처음 오는 숫자 다루려다가 코드가 정말 지저분해졌다..

처음엔 for문에 range 쓰기 귀찮아서 cnt넣고 한건데

그냥 range쓰는게 더 나을 뻔 했다.

 

소스코드

form = input()
elements = form.split('-')

result = 0
cnt = 0
for e in elements:
    if cnt is 0 and '+' not in e:
        result += int(e)
        cnt = 1
        continue
    if '+' in e:
        plus_elements = e.split('+')
        sum = 0
        for pe in plus_elements:
            sum += int(pe)
        if cnt is 0:
            result += sum
            cnt = 1
        else : result -= sum
    else :
        result -= int(e)

print(result)