Koala - 15기/기초 알고리즘 스터디

[백준/Python] 10773번: 제로

coouir 2024. 7. 11. 15:57

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

문제


문제풀이

정수를 K번 입력받는다. K번 정수를 입력받는 동안 정수가 "0"일 경우에는 pop()을 이용해서 정수를 저장한 리스트의 가장 마지막 값을 지운다. 정수가 "0"이 아닌 경우에는 append()를 이용해서 정수를 저장한 리스트에 값을 추가한다.

sum()을 이용해서 정수를 저장한 리스트에 있는 원소들의 합을 출력한다.


소스코드

K = int(input())

nList = []

for i in range(K):
    n = int(input())

    if n == 0:
        nList.pop()
    else:
        nList.append(n)

print(sum(nList))