파이썬(python) 백준 1541 : 잃어버린 괄호

---

백준 1541 : 잃어버린 괄호

---

문제 내용

시간 제한 : 2 초

메모리 : 128 MB

 

문제
양수와 +, -, 로 이루어진 식에 괄호를 적절히 쳐서 이 식의 값을 최소로 만들어라

입력
첫째 줄에 식이 주어진다.

식은 ‘0’~‘9’, ‘+’, ‘-’만으로 이루어져 있고, 가장 처음과 마지막 문자는 숫자이다.

그리고 연속해서 두 개 이상의 연산자가 나타나지 않고, 5자리보다 많이 연속되는 숫자는 없다. 

수는 0으로 시작할 수 있다. 입력으로 주어지는 식의 길이는 50보다 작거나 같다.

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

예제 입력 1 
55-50+40

예제 출력 1 
-35

 

유형

• 수학
• 그리디 알고리즘
• 문자열
• 파싱

---

풀이

이 문제는 그리디 알고리즘 문제이다.

괄호를 쳐서 식을 최소로 만들기 위해서는 최대한 -를 탐욕적으로 취해야 한다.

-1+1+1 은 -(1+1+1) 이렇게 괄호를 쳐야한다.

-1+1+1-1-1 은 -(1+1+1)-1-1 이렇게 괄호를 쳐야한다.

-가 1번이라도 등장하면 뒤의 + 값은 다 -로 처리해버리면 된다.

 

숫자가 0으로 시작할 수 있으므로 숫자를 tmp에 담아두었다가

+,- 연산자가 등장하면 int로 변환하여 결과값에 저장하였다.

int('001') = 1 이렇게 자동으로 변환되는 것을 이용한 것이다.

 

마지막으로 eval() 함수를 사용해서 문자열을 계산해내면 끝.

import sys
word = sys.stdin.readline().rstrip()
int_word = ''
tmp = ''
minus = False
for w in word:
    if w == '-':
        minus = True
        if tmp !='':
            int_word += str(int(tmp)) + w
        if int_word =='':
            int_word = w
        tmp = ''
    elif w == '+':
        if minus:
            w = '-'
        if tmp !='':
            int_word += str(int(tmp)) + w
        tmp = ''
    else :
        tmp += w
if tmp != '':
    int_word += str(int(tmp))
print(eval(int_word))

반례

입력 : -0-0

출력 : 0

 

입력 : -01-01

출력 : -02

 

입력 : -0001+0000

출력 : -1