[baekjoon]python #10773 제로

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

[baekjoon]python #10773 제로

10773번: 제로

처음에 시도했던 코드 -- 시간 초과!😥

n=int(input())
stack=[]
for i in range(n):
    temp = int(input())
    if temp !=0:
        stack.append(temp)
    else:
        stack=stack[0:len(stack)-1]

print(sum(stack))

두번째로 시도했던 코드 - 성공!

첫번째 코드에서 sys.stdin.readline으로 입력을 받았다. 하지만 5000ms라는 시간이 걸렸다.

import sys
input = sys.stdin.readline
n=int(input())
stack=[]
for i in range(n):
    temp = int(input())
    if temp !=0:
        stack.append(temp)
    else:
        stack=stack[0:len(stack)-1]

print(sum(stack))

세번째로 시도한 코드 - 정답!

128ms!

import sys
input = sys.stdin.readline
n=int(input())
stack=[]
sumV=0
for i in range(n):
    temp = int(input())
    if temp !=0:
        stack.append(temp)
        sumV+=temp
    else:
        sumV-=stack[-1]
        stack.pop()
print(sumV)

input받는 숫자를 temp에 넣어주고 0일때와 아닐때를 구분해 stack에 넣어주고 빼는 작업을 했다.

위의 코드들과 다른 점은 sumV를 통해 계산의 속도를 높였다는 점이다!

그래서 두번째 코드보다 약 1/25정도로 시간을 단축할 수 있었다.