Algorithm

[백준 1541] 잃어버린 괄호

choiht 2023. 1. 25. 12:09
반응형

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

 

1541번: 잃어버린 괄호

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

www.acmicpc.net

 

 

 

풀이

그리디 알고리즘이다. 

"-" 기호 사이에 있는 수, 또는 식을 괄호로 묶어주면 된다. 

 

"-" 를 기준으로 분리해서 n이라는 변수에 담아준다. 

0번째 원소는 무조건 양수이므로 answer에 더해준다. 

 

1번째 원소부터는 괄호로 묶이기 때문에 for문을 돌리며 answer에서 빼준다. 

 

 

 

코드

n = input().split("-")
answer = 0

for i in n[0].split("+"):
    answer += int(i)

for i in n[1:]:
    for j in i.split("+"):
        answer -= int(j)
    
print(answer)

 

 

반응형