[백준 1541] 문제풀이

문제

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

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

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

입력

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

출력

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

예제 입력 1 

55-50+40

예제 출력 1 

-35

예제 입력 2 

10+20+30+40

예제 출력 2 

100

예제 입력 3 

00009-00009

예제 출력 3 

0

 

정답

 

for문과 split의 적절한 배합과 알고리즘 문제 파악을 우선시 하며 풀어보자.

import sys
input = sys.stdin.readline

# 모든 입력값을 연산한 값을 저장해줄 변수 count
count = 0

# a에 입력된 값을 - 를 기준으로 나눠서 리스트에 문자열로 저장한다.
a = list(map(str, input().split("-")))

# for문을 사용해 식의 맨 처음값을 + 로 기준으로 나눈 뒤 i에 대입해서 루프를 돌린다. 
# ex) ['1+2+3+4'] => [1,2,3,4]
for i in a[0].split("+"):
    # count에 더해서 저장
    count += int(i)

# for 문을 사용해 -로 나눠진 리스트의 값을 i에 대입해서 루프를 돌린다.
# ex) ['1+2', '3+4'] 
for i in a[1:]:
    # 다시한번 for 문을 사용해 i의 값을 +를 기준으로 나눠준 뒤 j에 대입해 루프를 돌려준다.
    # ex) ['1+2', '3+4'] => [1,2,3,4]
    for j in i.split("+"):
        # 첫 + 연산 뒤의 모든 숫자는 다 -로 적용하여 계산하여도 무방하므로 다 - 연산 해준다.
        count -= int(j)
print(count)