Algorithm/백준

백준(Baekjoon) 1541번 잃어버린 괄호 문제 풀이 | 파이썬 Python, 그리디 알고리즘

daeunnniii 2021. 2. 24. 20:36
728x90
반응형

잃어버린 괄호

출처: www.acmicpc.net/problem/1541

 

1541번: 잃어버린 괄호

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

www.acmicpc.net

문제

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

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

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

입력

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

출력

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

예제 입력1

55-50+40

예제 출력1

-35

풀이

처음에 eval 함수를 사용해서 풀었다가 몇시간동안 헤어나올 수 없는 늪에 빠졌다.....

처음에 짰던 코드는 아래와 같다.

calculate=input().split('-')
result=0
result+=eval(calculate[0])
for i in calculate[1:]:
    result-=eval(i)
print(result)

IDLE로 코드를 실행해보았을 때, -가 없는 경우, 1개인 경우, 2개인 경우 다 결과값이 잘 나오는데

백준에 제출해보면 왜 런타임 에러(SyntaxError)가 나는 걸까??

계속 고민하며 몇시간은 코드를 수정해보았던 것 같다.........

 

근데 바로 그 이유는...!!

이 조건 때문이었다.

즉, 55-50+40을 입력하든 0055-050+000040을 입력하든 결과값이 잘 나와야하는데,

eval()은 앞에 0이 오는 수를 받아들이지 못한다.

 

 

따라서 eval()이 아닌 int()를 사용해서 다시 구현하였다.

eval()을 사용하지 못하므로 +를 기준으로 split()을 한번 더 해주어야한다.

calculate=input().split('-')
result=0
for i in calculate[0].split('+'):
    result+=int(i)
for i in calculate[1:]:
    for j in i.split('+'):
        result-=int(j)
print(result)

그 결과!! 맞았다..(^▽^)

728x90
반응형