본문 바로가기
TIL - 프로그래밍/Python 알고리즘

[백준] 10773. 제로 - Python

by chaemj97 2022. 6. 10.
728x90

https://www.acmicpc.net/problem/10773

 

10773번: 제로

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경

www.acmicpc.net


  • 코드1 - import
# 정수 K개
K = int(input())

num = []

for _ in range(K):
    n = int(input())
    # 0이 아니면 넣고
    if n != 0:
        num.append(n)
    # 0이면 직전에 넣은 거 버리김
    else:
        num.pop()
        
print(sum(num))
  • 코드2 - sys
from sys import stdin

K = int(stdin.readline())
total = []

for _ in range(K):
    n = int(stdin.readline())
    if n == 0:
        num.pop()
    else:
        num.append(n)

print(sum(num))
728x90
반응형

댓글