[Python] 백준/BOJ - 1541번: 잃어버린 괄호

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

1541번: 잃어버린 괄호

문제

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

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

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

 

입력

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

 

출력

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

 

예제 입력 1

55-50+40

 

예제 출력 1

-35

 

풀이

이 문제는 -뒤에 있는 수 끼리 더해서 앞에서 빼주면 된다. 55-(50+40)이나 55-(50+40)-(30+10) 처럼 생각하면 된다.

input_num = list(input().split('-'))
numbers = []

for i in input_num: # input_num=['55','50+40']
    s = i.split('+')
    cnt = 0
    for j in s:
        cnt += int(j)
    numbers.append(cnt)
result = numbers[0]

for i in range(1, len(numbers)):
    result -= numbers[i]
print(result)

(변수명 고민하다가 그냥 올립니다..)

먼저 문자열을 입력 받을 때 -를 기준으로 나눠 리스트에 저장해준다. 55-50+40 이라면 ['55', '50+40']으로 나눠진다.

 

numbers라는 이름의 빈 배열을 하나 선언해주고, 처음에 입력받은 배열을 for문을 돌며 한 요소씩 확인해준다.

요소 안에 +가 있으면 split해서 s에 저장해준 뒤 s 안에서 for문을 돌아 s 요소 안의 값을 cnt에 더해준 뒤 numbers 배열에 추가해준다.

 

input_num의 for문이 첫 번째 돌 때 (i=='55'일 때), 그 요소에 +가 없기 때문에 s는 '55'가 되고, cnt는 55가된다.

(현재 numbers=[55])

input_num의 for문이 두 번째 돌 때 (i=='50+40'일 때), 그 요소안에 +가 있기 때문에 s는 ['50', '40']이 되고, for문을 돌면서 그의 합 cnt는 90이 된다. (numbers=[55, 90])

 

55에서 90을 빼줘야 하기 때문에 result의 초기값은 numbers[0]인 55로 초기화해주고, 1부터 for문을 돌면서 numbers[i]의 값을 result에서 빼주면 된다.

 

---

다시 풀어보고 올리는 코드. 굳이 배열에 추가 안하고 바로 result에 값을 더하고 빼는 방식으로 바꿨다. 어찌풀든 답은 나온다.

input_num = list(input().split('-'))
res = 0
for i in range(len(input_num)):
    tmp = input_num[i].split('+')
    cnt = 0
    for x in tmp:
        cnt += int(x)
    if i == 0:
        res += cnt
    else:
        res -= cnt
print(res)