[백준]10773번 제로 - 스택

 

import sys
n = int(sys.stdin.readline())
stk = []
for i in range(n):
    value = int(sys.stdin.readline())
    if(value == 0):
        stk.pop()
    else:
        stk.append(value)
print(sum(stk))

 

코드 설명

먼저 반복할 횟수를 n 이라는 변수에 저장 후 빈 리스트를 생성한다.

값을 입력 받으면서 그 값이 0이라면 저장하지 않고, pop() 연산을 통해 이전에 받은 값을 지운다.

0이 아닐 경우에는 저장한다. sum() 함수를 이용하여 스택의 합을 출력한다.

 

Point : 0일 경우에는 저장 대신 삭제 기능을 구현한다.