최소힙/최대힙

import heapq # 1. 빈 리스트에 최소힙으로 item 추가 heap = [] heapq.heappush(heap, 50) heapq.heappush(heap, 10) heapq.heappush(heap, 20) print(heap) # [10, 50, 20] # 2. 기존리스트를 최소힙으로 만들기 heap2 = [50 ,10, 20] heapq.heapify(heap2) print(heap2) # [10, 50, 20] # 3. 가장 작은 원소(0번인덱스) 제거 result = heapq.heappop(heap) print(result) # 10 print(heap) # [20, 50] # 4. 그냥 작은 원소만 가져오고 싶다면 0번인덱스로 가져오기 result2 = heap[0]

heap_items = [1,3,5,7,9] max_heap = [] for item in heap_items: heapq.heappush(max_heap, (-item, item)) print(max_heap) # [(-9, 9), (-7, 7), (-3, 3), (-1, 1), (-5, 5)] # 실제 원소는 튜플의 1번 인덱스 값