백준[10773]python

문제 :

뭔가 길게 쓰여있지만... 그냥 0이 입력되면 마지막 값을 버리고 합을 출력하는 문제이다.

stack의 원리를 이용하면 아주 간단하게 해결할 수 있는 문제이다.

필자는 s라는 리스트를 만들고 stack처럼 사용했다. 0이 입력되지 않을 경우 s에 값을 모두 넣고 0이 입력되면 가장 위에 값(마지막으로 입력된 값)을 제거했다.

import sys
n = int(sys.stdin.readline())


if __name__=='__main__':

    s = []
    t = 0
    for i in range(n):
        tmp = int(sys.stdin.readline())
        if tmp != 0:
            s.append(tmp)
        else:
            s.pop()
    print(sum(s))