BAEKJOON_10773 "제로" PYTHON

[백준] 10773번 Python 파이썬

 

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

10773번: 제로

스택 알고리즘에 관한 문제이다!

 

money라는 리스트를 만들어서 a를 n개 입력받는다.

 

a가 0일 경우에는 money 리스트의 가장 마지막 인덱스 원소를 제거하고, 0이 아닐 경우에는 money에 원소를 담아준다.

 

가장 마지막 인덱스의 원소를 제거할 때는 pop() 함수를 사용한다.

import sys

n = int(sys.stdin.readline())
money = []

for i in range(n):
    a = int(sys.stdin.readline())
    if a == 0:
        money.pop()
    else:
        money.append(a)

print(sum(money))

혼자 공부하며 올리는 블로그입니다. 틀린 내용은 댓글 남겨주시면 감사하겠습니다.