백준, 제로, 10773

[유형]

자료구조, 스택

 

[문제링크]

https://www.acmicpc.net/problem/10773

 

[요약]

데이터를 입력받다가, 중간에 잘못된 데이터가 들어오면 0이 들어온다.

0이 들어오면 잘못된 데이터가 들어온 것이므로, 값을 삭제한다.

이후 남아있는 값들의 합을 구한다.

 

[문제 풀이]

FILO구조를 띄고 있는 자료구조로, 삽입과 삭제 연산이 동일한 곳에서 발생한다.

스택을 사용한다.

 

import sys

def input():
    return sys.stdin.readline().rstrip()

stack = []

N = int(input())

for _ in range(N):
    nn = int(input())
    if nn == 0:
        stack.pop()
    else:
        stack.append(nn)

print(sum(stack))