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

문제

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

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

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

입력

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

출력

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

 

 

---

 

풀이

쉽게 풀이를 생각할 수 있었다. 문자열을 하나씩 읽으면서, 숫자가 올 경우 저장한다(문자열로 하면 편하다. 다른 방법이 있을까?). 그리고 +나 - 연산 기호가 오는 경우,

• 처음 오는 숫자, 그리고 -기호가 나오기 전에 +가 나올 경우 저장된 숫자를 더해주고 해당 숫자를 초기화한다.
• 만일 -가 한 번이라도 나왔다면, 이후에 저장되는 모든 숫자는 빼줄 수 있다. 왜나하면, - 이후 +가 나오면 괄호로 묶어주면 이전의 - 연산에 의해서 빼는 효과를 낼 수 있다. 
• for문을 다 돈 이후, 저장되어있는 숫자는 -가 한 번이라도 나왔다면 빼주고, 아닌 경우 더해준다.

 

코드

string = input().strip()
flag = True
answer = 0
temp = ""

    for s in string:
        if flag:
            if s == "+":
                answer += int(temp)
                temp = ""
            elif s == "-":
                answer += int(temp)
                temp = ""
                flag = False
            else:
                temp += s
        
        else:
            if s == "+" or s == "-":
                answer -= int(temp)
                temp = ""
            else:
                temp += s

    if flag == True:
        print(answer+int(temp))
    else:
        print(answer-int(temp))

 

P.S.

제목도 괄호 하나를 잃어버렸다.