[10773] 제로

10773번: 제로

문제 그대로 0이 아닌 경우 리스트에 담아둔다. 0인 경우에는 nums에 추가를 해두어서 최근에 담은 것을 pop한다. pop을 하면 리스트에서 제거 되기 때문에 담은 리스트를 모두 sum 하면된다.

풀이

import sys

if __name__ == '__main__':
    K = int(sys.stdin.readline())
    nums = []
    for i in range(K):
        num = int(sys.stdin.readline())
        if num != 0:
            nums.append(num)
        else:
            nums.pop()

    print(sum(nums))