[백준] 1541 잃어버린 괄호 (파이썬)

1541번: 잃어버린 괄호

문제

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

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

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

입력

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

출력

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

예제 입력 1 복사

55-50+40

예제 출력 1 복사

-35

예제 입력 2 복사

10+20+30+40

예제 출력 2 복사

100

예제 입력 3 복사

00009-00009

예제 출력 3 복사

0

 

---

주제 - 그리디 알고리즘

 

식의 값을 최소로 만들기 위해서는 '-' 먼저 계산해야 한다. 그렇기 때문에 '-' 를 기준으로 괄호를 쳐주면 된다.

 

풀이과정

1. '-'를 기준으로 split한다.
2. 그러면 숫자로만 구성되거나 숫자와 '+'로 구성된 문자열이 생성된다.
3. 덧셈을 하기 위해서 그 문자열을 다시 '+' 기준으로 split하여 숫자만 더해서 num에 append 해준다. 
4. 식의 제일 처음은 숫자로 시작하기 때문에 num[0]번째 값에서 나머지 값들을 빼준다.

 

 

완성코드

import sys
input = sys.stdin.readline

n = input().split('-')

num = []

for i in n:
    sum = 0
    for j in i.split('+'):
        sum += int(j)
    num.append(sum)

n = num[0]

for i in range(1, len(num)):
    n -= num[i]
    
print(n)