그리디 알고리즘 : 잃어버린 괄호(백준 1541번)

▶ 문제 : https://www.acmicpc.net/problem/1541

1541번: 잃어버린 괄호

##### 문제 #####
# 세준이는 양수와 +, -, 그리고 괄호를 가지고 식을 만들었다. 
# 그리고 나서 세준이는 괄호를 모두 지웠다.
# 그리고 나서 세준이는 괄호를 적절히 쳐서 이 식의 값을 최소로 만들려고 한다.
# 괄호를 적절히 쳐서 이 식의 값을 최소로 만드는 프로그램을 작성하시오.

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

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

 

▶ 접근 방법

식의 값을 최소로 만들기 위해서는 첫 번째 '-' 뒤에 오는 값들의 합을 최대로 만들어야 했다.
이를 위해 '-'를 기준으로 식을 분리하고 해당 계산이 가능하도록 구현하였다. 

 

▶ 풀이 코드

import sys

# 입력 받은 식을 '-'를 기준으로 분리해 expression에 넣어준다.
expression = sys.stdin.readline().split('-')

result = 0
# expression의 0번 째는 첫 번째 -가 오기 전까지의 값이므로 
# 모두 result에 더해준다.
for ex in expression[0].split('+') :
    result += int(ex)

# 처음 -가 등장한 이후로의 값을 모두 확인하면서
for exp in expression[1:]:
    # 해당 값들을 result에서 빼준다.
    for ex in exp.split('+'):
        result -= int(ex)

print(result)

 

▶ 관련 링크

2021.11.17 - [알고리즘] - 이분 탐색(Binary Search)이란?