[백준] 10773번 제로 (Python 파이썬)

www.acmicpc.net/problem/10773

10773번: 제로

문제 설명

기본적인 스택을 이용한 문제이다.

첫째 줄에 입력할 정수의 개수 K를 입력하고 다음 K번 동안 정수를 입력받는다.

0일 경우 가장 최근에 넣은 수를 꺼내고, 나머지 경우에는 스택에 입력을 넣어주면 된다.

---

풀이 과정

0을 입력할 경우 가장 최근에 쓴 수를 지우라고 했으므로 스택의 pop 기능을 사용하면 된다.

 

마지막에는 sum함수를 이용해 스택의 모든 수를 더하면 된다.

k = int(input())
stack = []
for i in range(k):
    money = int(input())
    if money == 0:
        stack.pop()
    else:
        stack.append(money)

print(sum(stack))

 

https://github.com/HongEunho

전체 문제 & 코드는 위의 깃에 정리되어 있습니다.

팔로우 & 맞팔은 환영입니다 !