[1day 1sol] 최소힙

문제

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

1927번: 최소 힙

 

구현

• 만약 x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산 (x > 0)
     -. heapq.heappush() 메서드 이용
• 양수의 정수, 즉 자연수가 아니라면 heap list가 존재한다면 제거
     -. heapq.heappop() 메서드 이용
• 만약 리스트가 비어있는데 0이 나온다면 0출력

 

코드

import sys, heapq
input = sys.stdin.readline

n = int(input())
lst = list()

for _ in range(n):
    x = int(input())
    if x > 0:
        heapq.heappush(lst, x)
    else:
        if len(lst) != 0:
            print(heapq.heappop(lst))
        else:
            print(0)

 

느낀점

예외 처리를 해준다는 것이 중요하다는 것을 알게 되었다. 만약 리스트가 비어있는데 0이 들어오는 경우가 발생하는데 그 부분에 대한 생각을 해야한다는 점을 알게되었다. 항상 코드를 작성할 때, 만약 발생할 수 있는 예외 케이스를 생각하는 습관을 기르면 좋을 것 같다.