백준 알고리즘 1541. 잃어버린 괄호

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

1541번: 잃어버린 괄호

 

전체 코드

def solution(s: str):
    s_list = re.findall(r'\d+|[\+\-]', s)
    direction = 1
    result = 0
    for ch in s_list:
        if ch == '-':
            direction = -1
        elif ch == '+':
            pass
        else:
            result += int(ch) * direction

    return result


s = input()
print(solution(s))

 

정규표현식 패키지 re의 findall을 사용하여 ‘숫자’, ‘+’, ‘-’로 split를 진행한다. 최소값을 만드려면 1-2+3는 1-(2+3)으로, 1-2+3-4는 1-(2+3)-4으로 괄호를 묶을 수 있다.

이는 ‘-’가 나온 순간 부터 항상 뺄셈을 진행한다는 의미와 동일하다. 따라서 ‘-’가 나오면 항상 뺄셈을 진행하도록 코드를 작성한다.