새소식

Computer Science/코딩테스트 문제 풀이

[백준] [파이썬] [그리디] 1541번: 잃어버린 괄호

  • -

문제집 추천, 이 문제집에 나온 유형들만 공부해도 코딩테스트는 거뜬: https://www.acmicpc.net/workbook/by/BaaaaaaaaaaarkingDog

 

1541번: 잃어버린 괄호 문제보기

문제

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

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

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

입력

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

출력

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

예제 입력 1

55-50+40

예제 출력 1

-35

예제 입력 2

10+20+30+40

예제 출력 2

100


역시 쉽지 않다.

 

괄호가 없는 식이 주어졌다.

 

'-' 이후에 '+'가 나오면, '-' 뒤에 괄호를 쳤을때,
'-' 뒤에 (괄호로 묶은) 수들의 합이 음수가 되어 연산의 결과가 작아진다.
결국 '-'를 기준으로 해서 괄호를 치면 된다.

 

예를 들어 55-50+40 이 주어지면 55-(50+40)을 하면 최소값이 된다.

'-'를 기준으로 괄호를 쳐주면 되기 때문에 input을 받을 때 '-'를 기준으로 split 한다.
그러면 숫자로만 구성되거나 숫자와 '+'로 구성된 문자열들이 생성된다.
후에 각 문자열들을 다시 '+'를 기준으로 하여 split 하여 숫자만 남게 하고,
숫자들을 더해서 결과 값들을 다른 리스트에 저장해준다.

 

예를들어,

3 + 4 - 3 - 10 + 24
라는 식이 있을 때

 

1. '-' 기준으로 나누면,
'3 + 4'   '3'   '10 + 24'
이렇게 3개로 나눠진다

그러면 나눠진 각 문자열의 연산 결과를 알기 위해서

 

2. '+'를 기준으로 나눠 연산을 진행
7 3 34

 

3. 더해진 수들 사이에 마이너스 연산을 진행하면 된다.
7 - 3 - 34 = -30

 

4. 최종
(3 + 4) - (3) - (10 + 24) = -30
이렇게 괄호를 쳐서 연산한 결과와 동일한 결과가 나오게 된다.

 


 

정답

 

import sys

if __name__=="__main__":
    # 주어진 게 3+4-3-10+24 라하자 
    exp=input().split('-')  # ['3+4','3','10+24]
    nums=[]                 # '-'로 나눈 항들의 합을 저장할 리스트

    for e in exp:
        sum=0
        tmp=e.split('+')    # '3+4'를 더해줌 
        for j in tmp:
            sum+=int(j)
        nums.append(sum)    # 더한값 넣어줌 

    res=nums[0]             # nums에는 [7,3,34]가 들어 있음 7부터 시작해서 
    for i in range(1,len(nums)):    # 그 이후값들은 
        res-=nums[i]                # 전부 빼줌
    print(res)

 


 

Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.