[Python] 백준 1541 잃어버린 괄호 (Greedy)

📌 문제

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

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

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

입력

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

출력

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

예제 입력 1

55-50+40

예제 출력 1

-35

예제 입력 2

10+20+30+40

예제 출력 2

100

예제 입력 3

00009-00009

예제 출력 3

0

---

📌 풀이

💬 Code

import sys
input = sys.stdin.readline

data = input().rstrip().split('-')
ans = 0

for i in data[0].split('+'):
    ans += int(i)

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

print(ans)

💡 Solution

'-'가 나오면 그 다음 '-'가 나올 때까지 계속 빼 주었을 때 최솟값을 얻을 수 있다.
예를 들어,
40-10+30-50-60+70이 주어지면
40-(10+30)-(50+60+70)-20-80
이렇게 묶어주는 것! 따라서 '-'를 기준으로 나눠주기 위해 입력을 받고 split('-')을 진행한다.

40+20-0010+003-050-600+0700가 입력으로 들어왔다고 해보자. split('-') 후에는
['40+20', '0010+003', '050', '600+0700']가 data에 저장된다.

1. '가장 처음과 마지막 문자는 숫자이다.'라는 조건 때문에 첫번째 자리에 음수가 오는 경우는 없으므로 data[0]은 무조건 더해준다.

for i in data[0].split('+'):
    ans += int(i)

2. data[1], data[2], …는 '-'를 기준으로 나뉜 값이기 때문에 계속 빼준다.

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

• int('0010')하면 10으로 변환된다.
• 그리고 내장함수 중 eval()을 사용하면 string으로 표현된 계산식을 계산할 수 있다. 처음에 eval을 이용해서 풀려고 string에 괄호를 삽입하도록 작성했더니 0010같은 숫자 처리가 어려워져서 처음부터 뜯어고쳤다 ㅎ;

print(eval('40-(10+20)'))
# 10