백준 10773 제로 (파이썬)
접근방식
- 백준 10828 스택 문제를 풀었다면 쉽게 풀 수 있는 스택 개념 중 pop, push에 관련된 문제이다.
- 입력된 숫자가 0일때는 pop연산을 아닐때는 push연산을 하고 최종적으로 남은 배열의 sum을 구하는 코드를 작성하였다.
파이썬 코드
import sys
n = int(sys.stdin.readline())
arr = []
def push(n,arr):
arr.insert(0,n)
def pop(arr):
if len(arr) != 0:
arr.remove(arr[0])
for _ in range(n):
num = int(sys.stdin.readline())
if num == 0:
pop(arr)
else:
push(num,arr)
print(sum(arr))