Agorithm/백준 알고리즘
백준 알고리즘 17952번 과제는 끝나지 않아!(python)
kimjinho1
2021. 3. 6. 03:59
728x90
반응형
구현 & 스택 문제이다. 스택이 비어있을 때를 주의해서 잘 처리해줘야 한다. 런타임 에러로 엄청 틀렸다ㅜ
import sys
from collections import deque
stack = deque()
score = 0
for _ in range(int(sys.stdin.readline())):
li = list(map(int, sys.stdin.readline().split()))
if stack == [] or li[0] == 1:
stack.append(li[1:])
if stack:
stack[-1][1] -= 1
if stack[-1][1] == 0:
score += stack.pop()[0]
print(score)
728x90
반응형