python 算法实现 - 最小堆

最新推荐文章于 2025-07-06 17:34:08 发布

沧南

最新推荐文章于 2025-07-06 17:34:08 发布

阅读量3.5k

点赞数

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.csdn.net/Desilting/article/details/44963419

Python 专栏收录该内容

0 篇文章

订阅专栏

本文主要介绍如何使用Python语言实现最小堆数据结构，包括插入、删除、查找最小元素等核心操作，旨在帮助读者理解最小堆的工作原理及其在算法中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近浅学python，甚感方便，准备用python实现一遍常用的算法。

class MinHeap:
	def __init__(self):
		self.list = [0]
		self.size = 0

	def __str__(self) :
		return " ".join(str(v) for v in self.list[1:])
		
	def up(self, i) :
		while i / 2 > 0 :
			if self.list[i] < self.list[i / 2] :
				self.list[i], self.list[i / 2] = self.list[i / 2], self.list[i]
			i = i / 2

	def insert(self, k) :
		self.list.append(k)
		self.size += 1
		self.up(self.size)

	def minchild(self, i) :
		if i * 2 + 1 > self.size :
			return i * 2
		else :
			if self.list[i * 2] < self.list[i * 2 + 1] :
				return i * 2
			else :
				return i * 2 + 1

	def down(self, i) :
		while i * 2 <= self.size :
			minc = self.minchild(i)
			if self.list[i] > self.list[minc] :
				self.list[i], self.list[minc] = self.list[minc], self.list[i]
			i = minc

	def build(self, src_list) :
		i = len(src_list) / 2
		self.size = len(src_list)
		self.list.extend(src_list)
		while i > 0 :
			self.down(i)
			i -= 1

src_list = [9, 8, 7, 6, 5, 4, 3, 2, 1]

mh = MinHeap()
mh.build(src_list)
print mh

mh.insert(10)
print mh