[BaekJoon] 1541번: 잃어버린 괄호 (Python)

• 예전에 풀었던 문제인데 다시 풀었다.
• split 하는 방법만 알고 있으면 편함
  • 이 방법을 모르고 if문 사용한다면 빙빙 돌게 될 거임

Solution

import sys


def solution(s: str) -> int:
    answer = 0
    minus_s = s.split('-')

    for m in minus_s[0].split('+'):
        answer += int(m)

    for m in minus_s[1:]:
        for p in m.split('+'):
            answer -= int(p)

    return answer


s = sys.stdin.readline().rstrip()
print(solution(s))

Reference

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