본문 바로가기

IT/Python

[백준] 1541번 잃어버린 괄호 [Python] - 그리디

728x90

 

 

양수, +, -, 괄호가 주어진 수식에서 괄호만 지운 문자열이 주어질 때,

괄호를 적절히 쳐서 최소값을 구하는 문제

 

양수는 0부터 시작하는 문자열로 주어지기도 합니다.

 

더보기

문제

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

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

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

입력

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

출력

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

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

 

0으로 시작하는 문제를 해결하기 위해서 int를 사용했습니다.

 

split, 리스트 컴프리헨션, sum 을 사용한 코드

import sys
li = [sum([int(i) for i in s.split('+')]) for s in sys.stdin.readline().split('-')] + [0]
print(li[0]-sum(li[1:]))

 

split, map, lambda, join 을 사용한 코드

print(eval("-".join(map(lambda x: str(sum(map(int, x.split('+')))), input().split('-')))))

 

더보기

예제 입력 1 

55-50+40

예제 출력 1 

-35

예제 입력 2 

10+20+30+40

예제 출력 2 

100

예제 입력 3 

00009-00009

예제 출력 3 

0