[백준] 1541 잃어버린 괄호

문제 설명

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

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

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

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

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

예제 입력
55-50+40
예제 출력
-35

문제 풀이

실패한 풀이
'-' 문자 뒤로 오는 숫자들은 전부 음수로 계산하면 되기 때문에 무조건 음수가 되도록 괄호를 추가해주었다. 예를 들어 55-30+40-20 의 경우는 55-(30+40)-20 의 형태가 된다.

이렇게 식을 완성한다음 eval 함수를 이용해 식을 계산해 결과값을 구하려고 했다. 그러나 0009 와 같은 형식의 숫자를 eval 함수가 인식할 수 없었다. 그래서 re 모듈의 sub 함수와 정규표현식을 이용해 값을 인식가능한 형태로 변경시켜주었다.

그러나 백준은 python 의 외부 라이브러리를 사용할 수 없기 때문에 위와 같은 풀이를 사용할 수 없었다. 나름 효율적으로 푼 것 같았는데..

import re

str = input()
sol = "("

for c in str:
  if c == '-':
    sol += ")-("
  else:
    sol += c
    
sol += ")"
sol = re.sub("\\(0+", "(", sol)
print(sol)
print(eval(sol))