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

문제

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

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

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

입력

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

출력

 

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

---

풀이

[알고리즘]

괄호를 추가해서 최솟값이 나오기 위해서는 간단하게 생각하면 된다. 일반적으로 덧셈 과정을 모두 진행한 뒤 빼버리면 그 값이 최솟값이 된다. 파이썬에서 제공해 주는 split 내장 함수를 이용해서 간단하게 풀 수 있다.

 

[풀이 방법]

1. 입력 받은 문자열을 '-' 기준으로 쪼개준다.
2. 쪼개진 리스트의 ix가 0번째 값을 제외한 나머지들을 더한 뒤 더한 값을 0번째 값에 빼주면 답이 된다.
3. 위 과정을 예시로 풀어보면 아래와 같다.
4. 55-50+40 -> ['55', '50+40'] -> 55 - 90 -> -35  

import sys

ws = list(sys.stdin.readline().rstrip().split('-'))
b = 0
init = sum(map(int, ws[0].split('+')))
for w in ws[1:]:
    b += sum(map(int, w.split('+')))

print(init - b)

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

1541번: 잃어버린 괄호