백준 10773번. 제로 (Python / 파이썬)

문제

나코더 기장 재민이는 동아리 회식을 준비하기 위해서 장부를 관리하는 중이다.
재현이는 재민이를 도와서 돈을 관리하는 중인데, 애석하게도 항상 정신없는 재현이는 돈을 실수로 잘못 부르는 사고를 치기 일쑤였다.
재현이는 잘못된 수를 부를 때마다 0을 외쳐서, 가장 최근에 재민이가 쓴 수를 지우게 시킨다.
재민이는 이렇게 모든 수를 받아 적은 후 그 수의 합을 알고 싶어 한다. 재민이를 도와주자!

입력

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

출력

재민이가 최종적으로 적어 낸 수의 합을 출력한다. 최종적으로 적어낸 수의 합은 2^31-1보다 작거나 같은 정수이다.

---

문제 풀이

재현이가 재민이를 도와 돈을 불러주는 중인데, 잘못된 수를 부를 때마다 0을 외쳐서

가장 최근에 재민이가 쓴 수를 지우도록 해야합니다.

 

따라서 재현이가 부른 수를 저장할 list를 만들어주었습니다.

재현이가 수를 불렀는데 만약 0이라면

pop() 함수를 이용하여 list의 가장 오른쪽 원소를 삭제해주었습니다.

만약 0이 아니라면

append() 함수를 이용하여 list의 가장 오른쪽에 원소를 삽입해주었습니다.

 

모든 정수를 부를 때까지 이 과정을 반복하면 최종적으로 list에는

올바른 값들만 저장되게 됩니다.

따라서 sum() 함수를 이용해 list의 원소들의 합을 출력해주었습니다.

---

My Code

k = int(input()) # 정수의 개수

q = [] # 받아적은 수를 저장하는 list
for i in range(k):
    num = int(input()) # 재현이가 부르는 수
    if num == 0:       # 0을 외치면
        q.pop()        # list의 오른쪽에서 삭제
    else:              # 0이 아닌 수를 외치면
        q.append(num)  # list의 오른쪽에 삽입
    
print(sum(q)) # 받아적은 수들의 합을 출력