백준 10773 제로 (파이썬)


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))






© 2020.09. by 다로

Powered by theorydb