[BOJ: 1541] - Python / 파이썬 - 잃어버린 괄호

o_jooon_·2024년 1월 29일
0

BOJ

목록 보기
15/44
post-thumbnail

서론

그리디 문제입니다.
+와 -만 존재하기 때문에, 최대한 -(n1+n2)와 같이 묶어주면 최솟값을 만들 수 있습니다.
이를 위해서 -를 기준으로 문자열을 나누면 정답을 쉽게 구할 수 있습니다.

난이도

실버 2


문제

조건

시간 제한메모리 제한
2 초128 MB

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

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

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


입력

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


출력

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


예시

예제 입력 1

55-50+40

예제 출력 1

-35

예제 입력 2

10+20+30+40

예제 출력 2

100

예제 입력 3

00009-00009

예제 출력 3

0

풀이

짧은 코드로 이해하기 어려울 수 있어 예제 1번을 기준으로 풀이과정을 설명해드리겠습니다.

  1. 문자열을 입력받고 -를 기준으로 나눈다.
    55-50+40을 입력하면
    s = [55, 50+40] 이 된다.
  2. s의 각 인덱스에는 숫자와 +만 존재할 수 있기에 식을 모두 계산해준다(더해준다).
    ans = [55, 90]
  3. 문제의 조건에 따르면 입력 문자열의 처음과 끝은 숫자이기 때문에, 첫 숫자는 무조건 양수이다.
    따라서, 첫 숫자에서 뒤에 있는 숫자를 모두 빼주면 된다.
    55 - 90 = -35

코드

import sys
input = sys.stdin.readline

# 입력 문자열을 -를 기준으로 나눔
s = list(input().rstrip().split('-'))
# s에는 숫자 하나 또는 숫자 + 숫자와 같이 숫자 또는 +로만 구성이 되어있기 때문에,
# 각 인덱스에 존재하는 문자열을 다시 +를 기준으로 분리 한 후 모두 더하여 ans에 저장
ans = [sum(map(int, i.split('+'))) for i in s]

# 첫 숫자는 양수일 수 밖에 없기에 이후에 저장된 숫자들을 모두 뺄셈 후 출력
print(ans[0] - sum(ans[1:]))

실행 결과

profile
iOS개발 공부 중입니다.

0개의 댓글