[백준/Python] 1541번 잃어버린 괄호

1️⃣문제 설명

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

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

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

2️⃣입력

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

3️⃣출력

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

 

---

👩🏻‍💻작성한 코드

import sys

input = sys.stdin.readline

A = list(map(str, input().split("-")))
ans = 0


def Cal(s):
    tot = 0
    temp = str(s).split("+")
    for i in temp:
        tot += int(i)
    return tot


for i in range(len(A)):
    temp = Cal(A[i])
    if i == 0:
        ans += temp
    else:
        ans -= temp

print(ans)

 

💡코드 설명

수식의 값을 최소로 만들기 위해서는 가능한 큰 수를 빼야한다. 

 

문제에서 수식이 ' + ' 와 ' - ' 로만 이뤄져있기 때문에 더하기에 해당하는 부분을 괄호쳐서 그 값들을 모두 빼면 된다.

 

우선 들어온 문자열을 ' - ' 를 기준으로 쪼갠다.

 

그런 다음, Cal 함수에서 쪼개진 문자열들은 ' + ' 를 기준으로 쪼개고, 쪼개서 얻어진 데이터들을 합한 값을 반환합니다.

 

가장 앞에 있는 값만 ans에 더해주고 나머지 값들은 ans에서 빼준다.

 

ex) input : 420-30+150+32-5+16 

      A = ['420', '30+150+32', '5+16']

      temp = ['420']

                  ['30', '150', '32']

                  ['5', '16']