티스토리 뷰

728x90

문제

www.acmicpc.net/problem/10773

 

10773번: 제로

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

www.acmicpc.net

 

소스코드

n = int(input())

stack = []
num = [0]*n
for i in range(n):
    num[i] = int(input())

for i in range(n):
    if num[i] == 0:
        stack.pop()
    else:
        stack.append(num[i])

print(sum(stack))

for 문 두개인 코드. 이게 시간이 더 짧았다.

왜징 ???

 

n = int(input())

stack = []

for i in range(n):
    m = int(input())

    if m == 0:
        stack.pop()
    else:
        stack.append(int(m))

print(sum(stack))

for문 하나인 경우

반응형