[Python] 백준 1541번 잃어버린 괄호

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

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

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

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

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

예제 입력 1
55-50+40
예제 출력 1
-35
예제 입력 2
10+20+30+40
예제 출력 2
100
예제 입력 3
00009-00009
예제 출력 3
0

풀이과정

• 괄호를 특정 구간에 추가해서 결과값이 최소가 나오도록 하는 문제이다.
• 최소값이 나오게 하려면?
  연산자는 +,- 밖에 없다. 최대한 큰 숫자로 빼려면 - 를 기준으로 뒷부분은 빼기 연산만 하면 될 것 같다.
  • -가 없는 경우는?
    +만 있는 경우이다. 이는, 전부 다 더하면 된다.
    

num = input().split('-')
result = 0
 if len(num) == 1: 
     for i in num[0].split('+'):
         result += int(i)
     print(result)
else:
for i in num[0].split('+'):  # 배열의 첫번째는 더하기만 있다.
    result += int(i)
for i in num[1:]:  
    for j in i.split('+'): # 뒷 부분은 다 빼주면 된다.
        result -= int(j)
print(result)

-> 처음에는 +만 있는 경우를 고려해줬는데 코드가 유사한 부분이 겹쳐서 줄여도 같은 결과가 나올 것이라고 판단하여 수정해주었다.

제출 코드

num = input().split('-')
result = 0
for i in num[0].split('+'):  # 배열의 첫번째는 더하기만 있다.
    result += int(i)
for i in num[1:]:  # 뒷 부분은 다 빼주면 된다.
    for j in i.split('+'):
        result -= int(j)
print(result)