[Python] 백준 10773번 - 제로

문제

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

 

---

표준 입력을 위해 sys모듈 사용.

정수 하나 입력.

비어있는 리스트 생성.

import sys


input = sys.stdin.readline
k = int(input())
stack = []

 

입력받은 정수만큼 반복.

- 정수 입력.

- 입력 받은 정수가 0이 아니면 리스트에 추가.

- 아니면 제거.

리스트 요소의 합 출력.

for _ in range(k):
    n = int(input())
    if n:
        stack.append(n)
    else:
        stack.pop()
print(sum(stack))

 

전체 코드.

import sys


input = sys.stdin.readline
k = int(input())
stack = []
for _ in range(k):
    n = int(input())
    if n:
        stack.append(n)
    else:
        stack.pop()
print(sum(stack))