[python] 백준 10773 - 제로

정수 k가 주어지고

이후 k개의 줄에 정수가 주어진다.

k가 0일 경우 가장 최근 수를 지우고, 아닐 경우 해당 수를 쓰는 문제

 

import sys
k=int(sys.stdin.readline().rstrip()) #정수 k가 주어짐
li=[]
for i in range(k):
    a=int(sys.stdin.readline().rstrip())
    if a==0:
        del li[-1]  #0이면 최근 수 지움
    else:
        li.append(a)  #아니면 리스트에 추가
print(sum(li))

이것도 리스트로 풀었지만

큐로 다시 구현해봐야겠다.