[Greedy] 백준 잃어버린 괄호 1541 python3 풀이

https://www.acmicpc.net/problem/1541

1541번: 잃어버린 괄호

문제

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

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

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

입력

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

출력

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

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

 

import sys
inputs = sys.stdin.readline()
mIndex = inputs.find('-')
if (mIndex > 0 ) : 
    left = inputs[:mIndex]
    right = inputs[mIndex+1:].replace('+', '-')
    print(sum(map(int, left.split('+'))) + sum(map(lambda a: -int(a), right.split('-'))))
else:
    pIndex = inputs.find('+')
    if (pIndex):
        print(sum(map(int, inputs.split('+'))))


# inputs = "55-50+40"
# inputs = "55-50-40"
# inputs = "0550+40+ 1"
# inputs = "051"
# inputs = "0-1"
# inputs = "-1"
inputs = "0555+5-2-3"

mIndex = inputs.find('-')
if (mIndex > 0 ) : 
    left = inputs[:mIndex]
    right = inputs[mIndex+1:].replace('+', '-')
    print(sum(map(int, left.split('+'))) + sum(map(lambda a: -int(a), right.split('-'))))
else:
    pIndex = inputs.find('+')
    if (pIndex):
        print(sum(map(int, inputs.split('+'))))