프로그래밍/Python
[Python] 백준 10773 제로
모영이
2021. 2. 15. 18:05
10773번: 제로
첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경
www.acmicpc.net
스택은 파이썬의 배열로 구현이 가능하다.
1) 0이면 스택 팝
2) 0이 아니면 스택에 넣기
3) 스택의 합 출력
전체 코드
import sys
input = sys.stdin.readline
K = int(input())
stack = []
for _ in range(K):
a = int(input())
if a == 0:
stack.pop()
else:
stack.append(a)
print(sum(stack))