[백준_1541] 잃어버린_괄호_python

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

1541번: 잃어버린 괄호

문제

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

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

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

입력

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

문제풀이

괄호를 적절히 만들어 결과값을 최소로 만들어 출력하는 문제이다.

괄호를 적절히 사용가능하다의 의미는-가 나온 뒤의값들은 모두 음수라는 의미이다.

이 의미를 알기전엔 첫번째 코드와 같이 - 부호가 나올때마다 그 뒤에 나오는 값들에 대해 각가 부호처리를 해주었지만

-가 한번나오면 그 뒤의숫자들이 모두 음수라는것을 이해하면 split('-') 와 '+'를 따로 구분해서 입력받아 문제를 해결할 수 있다.

 

첫번째 코드는 정말 단순하게 + 혹은 -이 아니면 now에 값을 넣어주고 Int형으로 바꿔주어가며 계산하였다.

하지만 괄호의 의미를 좀 더 빨리 이해하신분들은 두번째와 같은 풀이방향을 생각했을것이다.

n = input()
answer = 0 #정답 축척공간
now = '' #문자형 숫자를 담을 공간
dir = 0 #짝수 -> 양수, 홀수 -> 음수
for i in n :
    if i == '+':    #+일때에는 dir에 따라서 answer값을 변경시켜줌
        if dir%2 == 0:
            answer += int(now)
            now = ''
        else:
            answer -= int(now)
            now = ''
    elif i =='-':   #-일때에는 dir이 짝수일때만 홀수로 바꿔줘
        if dir%2 == 0:
            answer += int(now)
            now = ''
            dir += 1
        else:       #-일때 dir은 유지
            answer -= int(now)
            now = ''
    else:
        now += i
if dir%2 == 0:      #마지막 숫자를 계산하기위해 dir에 따라서 answer에 값을 더해줘
    answer += int(now)
else:
    answer -= int(now)

print(answer)#마지막 숫자까지 더해주면 출력

-는 한번나오든 두번나오든 나온 그 순간부터 뒷값이 음수처리 가능하기때문에

우선 -로 나누어주어 b의 첫행만 양수처리하여 계산해주면 훨씬 깔끔한 식이 완성된다.

a = input().split('-')
b = [sum(map(int, i.split('+'))) for i in a]
print(b[0] * 2 - sum(b))