[자료구조/힙] 백준 1927 최소 힙 - Python

 

[ Contents ]

 

 

1. 문제 (링크 참조)

1927번: 최소 힙

 

 

 

2. 문제 풀이

 힙(Heap) 자료구조를 구현해서 푸는 문제입니다.

 

2022.02.20 - [Algorithm] - [Algorithm] 힙(heap), 최소/최대로 정렬하는 우선순위 큐

[Algorithm] 힙(heap), 최소/최대로 정렬하는 우선순위 큐

 힙 자료구조 설명과 구현 코드는 위 글에서 보실 수 있습니다.

 

 

 

3. 코드

import sys
import heapq
input = sys.stdin.readline

N = int(input())
heap = []

for _ in range(N):
    x = int(input())
    # x가 0이 아니면 push
    if x != 0:
        heapq.heappush(heap, x)

    # x가 0이고 비어있지 않으면 pop
    elif heap:
        print(heapq.heappop(heap))

    # x가 0이고 비어있으면 0 출력
    else:
        print(0)

 x가 0이 아니면 push하고, 0이고 비어있지 않으면 pop합니다.

 heap은 빈 리스트면 0(False)를 반환하고 비어있지 않으면 True를 반환합니다.