[백준/python] 1541번 : 잃어버린 괄호

문제

세준이는 양수와 +, -, 그리고 괄호를 가지고 식을 만들었다. 그리고 나서 세준이는 괄호를 모두 지웠다.

그리고 나서 세준이는 괄호를 적절히 쳐서 이 식의 값을 최소로 만들려고 한다.

괄호를 적절히 쳐서 이 식의 값을 최소로 만드는 프로그램을 작성하시오.

 

입력

첫째 줄에 식이 주어진다. 식은 ‘0’~‘9’, ‘+’, 그리고 ‘-’만으로 이루어져 있고, 가장 처음과 마지막 문자는 숫자이다. 그리고 연속해서 두 개 이상의 연산자가 나타나지 않고, 5자리보다 많이 연속되는 숫자는 없다. 수는 0으로 시작할 수 있다. 입력으로 주어지는 식의 길이는 50보다 작거나 같다.

 

출력

첫째 줄에 정답을 출력한다.

 

정답

import sys
input = sys.stdin.readline

exp = input()
num = ""
sign = "+"
answer = 0

minus = False
for i in exp:
    if i.isdigit():  # 숫자일 때
        num += i
    else:           # 숫자 끝
        num = int(num)

        if sign == "-":
            minus = True
            answer -= num

        elif sign == "+":
            if minus == True:   # 괄호 안
                answer -= num
            else:               # 괄호 밖
                answer += num

        sign = i    # + 또는 - 저장
        num = ""    # num 초기화


print(answer)

---

식의 값을 최소로 만들기 위해서는 최대한 많은 숫자를 빼야하기 때문에

-가 나왔을 때 괄호가 시작되어, 다음 -가 나올 때 괄호를 닫고 다시 괄호를 열어주어야 한다.

-가 나왔는지 확인하기 위해 기호를 저장할 sign을 만들어 초기값을 +로 설정해준다.

 

입력받은 식 exp는 문자로 이루어져있기 때문에 숫자를 분리해주어야 한다.

i가 숫자일 때 문자의 형태로 num에 더하고, i가 문자일 때 num을 숫자로 바꿔준다.

 

기호가 적용되는 숫자는 뒤에 오기 때문에, 기호를 sign에 저장해두었다가 다음 기호가 나왔을 때 적용해야한다.

 

만약 sign에 -가 저장되어있다면 minus가 나왔음을 알리기 위해 True로 바꾸고 answer에서 num을 뺀다.

sign에 +가 저장되어있을 때, minus가 True라면 괄호 안을 의미하므로 answer에서 num을 뺀다.

sign에 +가 저장되어있을 때, minus가 False라면 괄호 밖을 의미하므로 answer에서 num을 더한다.

 

sign에 다음 문자를 저장하고 num을 빈 문자열로 다시 초기화 해준다.