본문 바로가기

알고리즘

[알고리즘] 제로(백준 10773/파이썬)

[개념]스택

 

[문제]

 

 

[문제해결방식]

-입력받은 숫자를 더하는데 중간에 0이 들어오면 그 전값을 지워 버리고 다시 더하기 시작 한다.

[코드리뷰]

import sys

k=int(sys.stdin.readline())
A=[]      
#리스트에 저장하기

for i in range(k):
    n = int(sys.stdin.readline())
    if n == 0:
        A.pop()
        #0이 들어오면 그전에 최근 수를 지운다
    else:
        A.append(n)
        #0이 아니라면 받은 값을 리스트에 저장한다
print(sum(A))
#더한 값을 출력한다

[결과]