[백준] 10773번 제로 (python)

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

10773번: 제로

 

 

재민이와 재현이의 역할을 잘 파악하면 된다.

재민 : 장부 관리

재현 : 잘못된 수를 외침 → 이로 인해 재민이는 가장 최근에 쓴 수를 지움

 

문제 해결 방법

재현이가 0을 외치면 스택 마지막 요소를 삭제한다.

0을 외치지 않으면 제대로 부른 것이니, 해당 숫자를 스택에 추가한다.

 

소스 코드

import sys
input = sys.stdin.readline

N = int(input())
stack = []
for _ in range(N):
    number = int(input())
    
    if number != 0:
        stack.append(number)
    
    else:
        stack.pop()

print(sum(stack))