BOJ / Implementation / 제로 / Python

<10773번>

문제 풀이

k번 숫자를 입력받고 0이라면 정수리스트.pop(), 0이 아니라면 정수리스트.append(정수)를 해준다.

코드

import sys
input = sys.stdin.readline

k = int(input())
nums = []

for _ in range(k):
    num = int(input())
    if num == 0:
        nums.pop()
    else:
        nums.append(num)

print(sum(nums))

📝 문제 풀면서

1. sys.stdin.readline()
   한 줄 단위로 입력받으며 개행문자까지 입력받는다.