[백준] 1541번 잃어버린 괄호 - Python / 알고리즘 중급 1/3 - 그리디 알고리즘 (연습)

---

---

---

💡 문제

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

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

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

입력

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

출력

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

---

💭 접근

이 문제의 경우 괄호를 치는데에 제한이 없기 때문에, 처음 - 연산자가 등장하면 이 이후의 숫자들은 모두 뺄 수 있다.

따라서 입력으로 주어진 식의 값을 최소로 만들기 위해선, 처음으로 등장한 - 연산자 이전의 숫자를 모두 더한 값에서 - 연산자 이후의 숫자를 모두 빼주면 된다.

---

📒 코드

s = input()

if len(s) == 1:
    print(int(s))
else:
    num = []
    oper = []

    tmp = ''
    for i in range(len(s)):
        if s[i] == '+' or s[i] == '-':
            oper.append(s[i])
            num.append(int(tmp))
            tmp = ''
        else:
            tmp += s[i]
    num.append(int(tmp))

    for i in range(len(oper)):
        if oper[i] == '-':
            print(sum(num[:i + 1]) - sum(num[i + 1:]))
            break
    else:
        print(sum(num))

---

💭 후기

오랜만에 쉬운 문제라서 가벼운 마음으로 풀었다.

---

🔗 문제 출처

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

---