파이썬 그리디 알고리즘 백준 1541 잃어버린 괄호

문제

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

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

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

입력

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

출력

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

풀이

연산의 합이 최소가 되게 하려면 +기호로 묶인 두 수를 묶어 -기호와 연산될 때 가장 큰 값이 빼지게 하면 된다.

따라서 -기호로 split하여 입력값을 나눠주고, 나눠진 입력값을 +기호로 split 하여 더해준다.

이후 첫번째 값 sums[0]에 이후의 값들을 빼주면 최소값을 구할 수 있다.

예제로 예를 들면,

55-50+40를 -로 split 하면 [55, 50+40] 를 구할 수 있고

이것을 +기준으로 split하여 더해주면 [55, 90] 을 구할 수 있다.

이제 -로 나눴던 값을 빼주면 55-90이 되어 -45라는 값을 구할 수 있다.

import sys
def answer(nums):
    sums = []
    for num in nums:
        temp = list(map(int, num.split("+")))
        sums.append(sum(temp))

    total = sums[0]
    for i in range(1,len(sums)):
        total -= sums[i]
    return total

if __name__ == "__main__":
    nums = sys.stdin.readline().rstrip("\n").split("-")
    print(answer(nums))