[백준] 10773번 제로 (파이썬)

│문제

10773번: 제로

│풀이

K=int(input())
stack=[]
for i in range(K):
    num=int(input())
    if num==0:                  #pop연산
        stack.pop(len(stack)-1)
    else:                       #push연산
        stack.append(num)

sum=0
for i in range(len(stack)):sum+=stack[i]
print(sum)

│설명

 

문제는 스택의 구조와 동일하다.

0이아닌 정수는 push ,0인 경우에는 pop연산을 실행한다.

풀이 코드의 len(stack)-1은 스택의 탑을 의미한다.