algorithm/baekjoon

[백준] 1541번 잃어버린 괄호(파이썬)

support_u 2023. 3. 30. 04:38

문제

 

1541번: 잃어버린 괄호

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

www.acmicpc.net

코드

n = input().split("-")
li = []

for i in n:
    num = ""
    if "+" in i:
        m = i.split("+")
        num = 0
        for j in m:
            num += int(j)
        li.append(num)
    else:
        li.append(int(i))

if len(li) > 1:
  num = li[0]
  for i in li[1:]:
      num += -i
  print(num)
else:
    print(*li)

해설

split으로 '-', '+' 나누고 int로 바꿔서 더해주고 list에 넣어주고 list에 2개 이상 있다면 음수로 빼주는 형식으로 작성하였다.