백준 10773 (python): 제로

 

풀었던 문젠데 복습 겸 다시 풀어봤다.

코드가 훨 간결해진 듯

 

🚨 이번에 푼 것

import sys

k = int(input())
stack = []
for i in range(k):
    num = int(sys.stdin.readline().strip())
    if num == 0:
        if stack:
            stack.pop()
    else:
        stack.append(num)

print(sum(stack))

 

🚨 옛날에 푼 것

K = int(input()) #반복 횟수 입력
numList = []
for i in range(0, K):
    num = input()
    if num == "0":
        numList.pop(-1)
    else:
        numList.append(num)

hap = 0
for i in numList:
    hap += int(i)

print(hap)

❓ 근데 이거 왜 시간초과 안 났지? 아 pypy로 제출하니까 시간초과 안 난당