개발공부/algorithm

[BOJ] [Python] 백준 자료구조 - 10773: 제로

so.py 2020. 10. 6. 16:28

문제: 10773: Zero

 

10773번: 제로

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경

www.acmicpc.net

접근

기본 스택 자료구조를 사용하면 쉽게 풀 수 있는 문제다. 입력 값에 0이 들어오면 stack.pop, 아니면 append를 해주면 된다. (이런 쉬운 문제만 계속 나왔으면 좋겠다 ㅎㅎ)

내 코드

T = int(input())
stack = []
for i in range(T):
    N = int(input())
    if N != 0:
        stack.append(N)
    elif N == 0:
        stack.pop()
print(sum(stack))