[solved.ac 실버2] 1541_잃어버린 괄호

제한 사항

문제

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

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

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

입력

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

출력

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

예제

---

나의 풀이

연산자로 + 와 - 만 오기 때문에 -의 값이 최대가 되면 최소값을 구할 수 있다.

예를 들어 "3+2-9+1+2+3+4+5-6+9" 라는 식이 있으면 "3+2 - (9+1+2+3+4+5) - (6+9)" 와 같이 묶으면 최소값을 만들 수 있는 것이다.

나도 처음엔 eval 함수를 떠올렸다. eval 함수는 수식을 그대로 계산할 수 있다. (사실 그보다 더 많은 기능이 있지만 여기서 설명은 하지 않는다.)

그러나 문제의 조건 중 0으로 시작하는 숫자도 올 수 있다고 되어있다. 하지만 eval 함수는 012+2 를 12+2처럼 계산하지 못하고 오류를 발생시킨다.

그래서 eval 함수를 사용하지 않고 직접 계산하면 된다.

'-' 를 기준으로 묶으면 '+'만 남게 되는데 그냥 '+'로 다시 묶어서 숫자들을 전부 더해버리면 된다. (sum 함수가 떠올랐다면 굿)

def solution(expression):
    calc = lambda exp: sum(map(int, exp.split('+')))
    expressions = expression.split('-')
    answer = calc(expressions[0])
    if len(expressions) == 1: return answer
    for e in expressions[1:]:
        answer -= calc(e)
    return answer
print(solution(input()))

나는 calc라는 함수를 선언했다. calc 함수는 '+'만 가진 식이 매개변수로 주어지면 '+' 기준으로 쪼개서 나온 숫자(타입은 문자)를 정수형으로 바꿔서 sum 함수로 합을 구하는, 즉 '3+2+1' 과 같은 식이 매개 변수로 주어지면 6을 리턴하는 함수이다.

가장 먼저 expression을 '-'로 쪼갠다. 그러면 맨 앞 숫자는 '+' 이고 나머지는 '-'로 묶인 것이다.

만약 식에 '-'가 없다면 expressions을 '-' 기준으로 쪼갠 결과 리스트인 expressions의 길이가 1이다. 이때 뒤에서 expressions[1: ] 을 하면 인덱스에러가 발생하기 때문에 만약 '-'가 없는 경우에는 바로 return 하고 끝낸다.

'-'가 식에 있다면 expressions의 길이는 2 이상일 것이다.

1번 인덱스 이후부터는 전부 빼면 된다.

채점 결과