백준 온라인 저널, 그리디 알고리즘/1541번 : 잃어버린 괄호(파이썬)

 

문제 정의

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

 

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

 

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

 

입력

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

 

출력

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

 

예제 입력 1

55-50+40

예제 출력 1

-35

 

 

접근 방법

0. 값을 최소화하기 위해선 -로 묶이는 괄호의 값이 가장 크면 된다. 따라서 괄호의 값을 가장 크게 하기 위해 - 뒤에 있는 +로 묶을 수 있는 숫자는 모두 괄호로 묶고 -가 나오면 괄호를 다시 묶는다.

 

1. 우선 '-'로 나눠 리스트로 만든다.

2. 해당 리스트를 하나하나씩 뽑아서 '+'를 기준으로 또 나눠 리스트를 만들고, 숫자는 모두 정수형으로 바꾸고 이 값을 빼준다.

 

코드

data = input()
d = data.split('-')

for i, x in enumerate(d):
    bracket_number = 0
    for x_ in x.split('+'):
        bracket_number += int(x_)
    if i == 0:
        result = bracket_number
    else:
        result -= bracket_number

print(result)

 

어려웠던 점

- 숫자의 처음이 0이 나올수도 있다는 조건(ex. 03)때문에 eval 함수가 먹히지 않았다. 더 간단하게 처리할 수 있을 것이라 생각했지만 해당 조건을 확인하지 않아서 자꾸 에러가 났다. 앞으론 문제와 입력, 출력을 모두 잘 확인해야겠다.