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




문제





나의 답안

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

스택 문제이다. 스택은 FILO(First In Last Out) 구조이다. pop을 통해 가장 마지막 값을 뺄 수 있으며, push를 통해 값을 추가할 수 있다.
1. 반복문을 통해 k개수만큼의 입력값을 받는다.
2. 만약 n이 0이면 스택에서 마지막 인자를 제거한다.(pop)
3. 아니라면 스택에 append메소드를 이용해 입력받은 값을 추가해준다.(push)

좋은 웹페이지 즐겨찾기