1927:최소힙

22 Dec 2022 in PS on Ps

문제

코드

import sys
import heapq

N = int(sys.stdin.readline())
data = []

for i in range(N):
    a = int(sys.stdin.readline())
    if a != 0:
        heapq.heappush(data, a)
    else:
        if not data:
            print(0)
        else:
            print(heapq.heappop(data))

풀이

파이썬 라이브러리 heapq를 사용

후기

힙 자료구조에 대해 다시 한번 복습 할수 있어 좋았다.
파이썬 라이브러리는 최소 힙으로 구현이 되어있기 때문에 최대힙을 구현 하기 위해서는 - 부호를 활용하면 된다.