[백준] #1541 잃어버린 괄호 (python)

https://www.acmicpc.net/problem/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

---

# Code 

sol1

import sys
input = sys.stdin.readline

s = input().split('-')
num = []
for i in s:
    sum=0
    a = i.split('+')
    for j in a:
        sum+=int(j) # int타입으로 바꿔서 연산
    num.append(sum)

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

 

sol2

import sys
input = sys.stdin.readline

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

sum = 0
for i in s[0].split('+'):
    sum+=int(i) # int타입으로 바꿔서 연산

for i in s[1:]:
    for j in i.split('+'):
        sum-=int(j)

print(sum)

# Comment

연산의 값이 최소가 되려면 '-'가 나왔을 때 최대한 큰 값을 빼주도록 하면 된다. 즉, 55-50+40-20+60 의 경우 55-(50+40)-(20+60) 이런식으로 '-' 뒤에 또 다른 '-' 값이 오기 전까지 다 묶어서 연산해주면 각각의 '-' 마다 최대의 값을 뺄 수 있으므로 연산의 값이 최소가 된다. 

예를 들어서 코드를 설명해보겠다. 만약 55-50+40-20+60 이 input 값으로 주어진다면, '-'를 기준으로 값을 잘라서

s = [ '55', '50+40', '20+60' ] 가 된다.

sol1의 경우 여기서 50+40과 20+60의 값을 각각 sum 변수에 저장해서 num 리스트에 담는다. 따라서 num = [ '55', '90', '80' ] 이 되고, result 변수에 num[0] 값인 55를 넣어준 뒤 뒤의 원소들을 차례로 빼주는 방식이다.

sol2의 경우 따로 num 리스트를 사용하지 않고, sum 변수가 sol1의 result 변수 역할까지 한번에 수행한다. s[0]의 값인 55를 sum에 먼저 더해주고, 그 뒤에 오는 원소들은 차례로 sum에서 빼주는 방식이다. 즉, ((((55-50)-40)-20)-60) 의 형태로 계산이 된다.

 

* sum+=int(i) : s의 원소 타입이 문자이기 때문에 각 원소를 int 타입으로 바꿔준 뒤 연산해야 한다.