본문 바로가기
개발/Python

파이썬 python 자료구조 힙 heapq 간단 구현

by SPNK 2023. 1. 31.
반응형
  • 최소 힙으로 변환
import heapq

# 목록을 최소 힙으로 변환
my_list = [4, 2, 1, 3, 5]
heapq.heapify(my_list)
print(my_list)
# 출력: [1, 2, 3, 4, 5]

 

  • 힙에 요소 삽입
# 힙에 요소 삽입
heapq.heappush(my_list, 0)
print(my_list)
# 출력: [0, 2, 1, 4, 5, 3]

 

  • 힙에서 가장 작은 요소 팝
# 힙에서 가장 작은 요소 팝
smallest = heapq.heappop(my_list)
print(smallest)
# 출력: 0
print(my_list)
# 출력: [1, 2, 3, 4, 5]

 

  • 힙에서 가장 작은 요소 찾기
# 힙에서 가장 작은 요소 찾기
smallest = my_list[0]
print(smallest)
# 출력: 1
반응형

댓글