자료구조 스택
작성 완료
- 자료를 넣고(push) 자료를 빼는(pop) 입구가 같은 선형 구조(LIFO - Last In First Out)
- 참고: 자료구조 스택
- 파이썬에서 스택은 list를 통해 구현할 수 있다
- stack.append(x)를 통해 스택에 x를 오른쪽(뒤)에 push한다
- stack.pop()을 통해 스택의 마지막 원소를 pop한다
def solution():
K = int(input())
stack = []
for _ in range(K):
num = int(input())
if num == 0:
stack.pop()
else:
stack.append(num)
return sum(stack)
print(solution())