10773번 제로

문제링크 : www.acmicpc.net/problem/10773

10773번: 제로

 

---

---

<알고리즘>

  1) 입력받는 숫자가 0이 아니라면 배열의 첫번째에 삽입

  2) 입력받는 숫자가 0이라면 가장 최근 삽입된 요소를 지우는 pop함수 실행(단 최근 요소 출력X)

 

 

버전 1. 함수 이용(  메모리 : 29556KB    시간 : 2608ms )

numbers = []
def pop():
    del numbers[0]
import sys
N = int(sys.stdin.readline())
for _ in range(N):
    num = int(sys.stdin.readline())
    if(num==0):
        pop()
    else:
        numbers.insert(0,num)
result =0
for i in range(len(numbers)):
    result+=numbers[i]
print(result)

 

버전 2. main안에서 한번에 처리 ( 메모리 : 29556KB    시간 : 2552ms)

import sys
N = int(sys.stdin.readline())
numbers = []
for _ in range(N):
    num = int(sys.stdin.readline())
    if(num == 0):
        del numbers[0]
    else :
        numbers.insert(0,num)
result =0
for i in range(len(numbers)):
    result += numbers[i]
print(result)