从 O(n log n) 到 O(n)深入理解 Hello Algo 中的堆构建Heapify与复杂度推导【免费下载链接】hello-algo《Hello 算法》动画图解、一键运行的数据结构与算法教程。支持简中、繁中、English、日本語提供 Python, Java, C, C, C#, JS, Go, Swift, Rust, Ruby, Kotlin, TS, Dart 等代码实现项目地址: https://gitcode.com/GitHub_Trending/he/hello-algo本文聚焦《Hello 算法》hello-algo堆章节的建堆操作Heap Construction系统对比逐个插入建堆与自底向上堆化建堆两种实现路线并结合仓库源码如 Python 版 my_heap.py、C 版 my_heap.cpp与原文档 build_heap.md 中的完整数学推导说明为什么基于堆化遍历的建堆方法能将复杂度从 $O(n\log n)$ 优化到 $O(n)$。阅读后你将掌握数组堆的建堆原理、正确确定最后非叶节点的方法以及用错位相减法严谨推导 $O(n)$ 建堆复杂度的完整过程。背景什么是堆与建堆操作堆Heap是一种满足特定条件的完全二叉树主要分为两类详见 堆基础章节 heap.md小顶堆min heap任意节点值 $\leq$ 其子节点值大顶堆max heap任意节点值 $\geq$ 其子节点值。由于堆是一种完全二叉树非常适合用数组表示数组元素即节点值、数组下标即节点位置父子关系由下标映射公式确定——节点 $i$ 的左子为 $2i1$、右子为 $2i2$、父节点为 $(i-1)/2$向下取整。在工程实践中priority queue 与heap往往被看作等价概念堆即优先队列的典型底层实现。本文与后续内容均以大顶堆为例小顶堆只需将所有比较符号反向即可源码注释中也对此作了明确说明。**建堆heap construction**是指给定一个包含全部元素的无序列表通过特定流程使其满足堆性质最终得到一棵合法的堆。本文关联文档给出了两种做法及其复杂度对比下文依次展开。方法一逐个插入建堆 —— 直观但代价较高思路非常朴素先创建一个空堆然后遍历列表对每个元素依次执行一次元素入堆操作。入堆意味着先把元素追加到堆底再对该元素执行一次自底向上的堆化heapify。每插入一个元素堆的长度加一由于节点按从上到下、从左到右的顺序补入完全二叉树堆整体是自顶向下长起来的堆化的路径长度为树高即每次插入耗时 $O(\log n)$对 $n$ 个元素逐一执行总时间复杂度为 $O(n\log n)$。若以 my_heap.py 中的push()/sift_up()为插入原子操作那么逐个插入建堆就是把这两个方法在循环里重复 $n$ 次。这也是大多数语言内建优先队列的默认增量行为。方法一易于理解、无需额外空间但当构建目标是由现有列表一次性成堆时其 $O(n\log n)$ 并非最优。方法二堆化遍历建堆 —— 两步实现 O(n) 构建原文档给出的高效建堆只需两步原样装入直接把列表全部元素按顺序放入堆数组此刻堆性质大概率尚未满足逆序堆化按层序遍历的反向顺序遍历对每个非叶节点依次执行一次自顶向下堆化sift down。其正确性依赖一条关键不变量对某节点完成堆化后以该节点为根的子树就成为一个合法的小堆子堆。由于采用逆序遍历轮到当前节点时其下方子树已经是合法子堆此时对它做下沉堆化才真正有效堆由此自底向上被构建。一个值得强调的推论是叶节点天然是合法子堆无需堆化。因此遍历的起点不是数组末尾而是最后一个节点的父节点也就是最后非叶节点。据此可以从数组末尾反推出起点下标Python 实现见 my_heap.py__init__中先整体赋值self.max_heap nums再从self.parent(self.size() - 1)逆序sift_down(i)到根def __init__(self, nums: list[int]): Constructor, build heap based on input list # Add list elements to heap as is self.max_heap nums # Heapify all nodes except leaf nodes for i in range(self.parent(self.size() - 1), -1, -1): self.sift_down(i)C 的对应逻辑见 my_heap.cpp构造器内maxHeap nums;后执行for (int i parent(size() - 1); i 0; i--) siftDown(i);。下沉sift_down本身需要比较当前节点与其左右孩子的大小必要时与较大的孩子交换并继续下探Python 版实现位于 my_heap.py终止条件为越过叶节点或当前节点无需修复。为什么逆序且只处理非叶节点仓库源码给出了两种语言中完全一致的写法可以相互印证起点parent(size() - 1)—— 最后一个元素堆底最右叶节点的父节点即全堆最后一个非叶节点方向从该下标递减到0根正好是层序遍历的反序范围不包含叶节点下标区间从而省去无意义的堆化调用。C 与 Python 版唯一的工程差异在于C 通过 vector 以动态数组存储以避免扩容问题Python 直接复用传入的list。核心算法完全同构。复杂度分析直观估算为何不准确直观但不准确的估算设完全二叉树有 $n$ 个节点叶节点数量约为 $(n1)/2$向下取整因此需要堆化的非叶节点数量约为 $n/2$自顶向下堆化中每个节点最多下沉到叶节点故单个节点最大迭代次数约等于树高 $\log n$。两者相乘得到 $O(n\log n)$。但原文档明确指出现这个估算并不准确——它忽略了一个重要事实完全二叉树中越靠近底层的节点数量越多底层的大量节点下沉距离很短甚至为 0如果一律按树高 $\log n$ 估算会显著高估总工作量。精确推导按层累加工作量为简化推导假设考察一棵有 $n$ 个节点、高度为 $h$ 的完美二叉树该假设不影响结论正确性。如原文档配图所示图中展示了推导的两条核心事实节点高度右侧标注某节点自顶向下堆化的最大迭代次数恰等于它到叶节点的距离也就是该节点的高度根高度为 $h$逐层递减叶节点高度为 0无需堆化恰好落在被跳过的区间每层节点数右侧标注第 $k$ 层自根起 0 层计数节点数为 $2^k$——根层 $2^01$第二层 $2$第三层 $4$以此类推叶层 $2^h$。于是全部节点的堆化迭代总次数 每一层的节点数 × 节点高度之和$$ T(h) 2^0h 2^1(h-1) 2^2(h-2) \dots 2^{(h-1)}\times1 $$注意公式中求和不包含高度为 0 的叶节点层恰好与源码只堆化非叶节点的范围一致。错位相减把 T(h) 化为等比数列对 $T(h)$ 做标准的高中数列处理——先整体乘以 $2$ 得到错位一行的 $2T(h)$$$ \begin{aligned} T(h) 2^0h 2^1(h-1) 2^2(h-2) \dots 2^{h-1}\times1 \newline 2 T(h) 2^1h 2^2(h-1) 2^3(h-2) \dots 2^{h}\times1 \end{aligned} $$用第二式 $2T(h)$ 减去第一式 $T(h)$即移位相减/错位相减法中间项成对消去$$ 2T(h) - T(h) T(h) -2^0h 2^1 2^2 \dots 2^{h-1} 2^h $$观察上式$T(h)$ 的主体 $(2^12^2\dots2^h)$ 是一个等比数列直接用求和公式即可算出其量级$$ \begin{aligned} T(h) 2 \frac{1 - 2^h}{1 - 2} - h \newline 2^{h1} - h - 2 \newline O(2^h) \end{aligned} $$进一步地高度为 $h$ 的完美二叉树共有 $n 2^{h1} - 1$ 个节点于是$$ O(2^h) O(n) $$结论基于逆序堆化遍历的建堆方法其时间复杂度为 $O(n)$远优于逐个插入的 $O(n\log n)$。这就是两种建堆方法在复杂度上的本质差异——看似都在做堆化但堆化的总代价因为按层累加而大幅收敛到线性。复杂度结论在源码中的体现与延伸上述 $O(n)$ 建堆并非只在理论推导中存在仓库的工程实现同样可以印证Python 的标准库在 heap.py 之外还提供heapq.heapify()完成原地线性建堆heap.md 给出的调用示例为heapq.heapify(min_heap)对[1, 3, 2, 5, 4]就地成堆C 侧也可直接用范围构造priority_queueint, vectorint, greaterint minHeap(input.begin(), input.end())一次成堆本仓库自实现的MaxHeap.__init__/MaxHeap(vectorint)正是两步堆化建堆算法的直接落地可运行驱动代码位于同一文件的if __name__ __main__/main()中my_heap.py会打印建堆后的数组与树形表示便于读者核对结果。建堆 $O(n)$ 的高效率是堆结构在多个经典场景中被广泛采用的基石例如出处同为堆章节优先队列入队、出队均为 $O(\log n)$建堆为 $O(n)$整体吞吐可观堆排序先线性建堆、再反复取出堆顶即可得到有序序列另有更精巧的就地实现见堆排序章节Top-k 问题维护一个大小为 $k$ 的小顶堆来筛出前 $k$ 大元素仓库配套实现见 top_k.py 及其对应的文档 top_k.md。若希望亲手验证逐节点插入与线性建堆两套路线的行为差异可对照仓库中 Python 源码 与 C 源码 进行运行观察。heap.md 中对数组表示、push/pop/peek/size/isEmpty等操作效率的完整表格详见堆基础章节 heap.md是理解本文建堆上下文的前提建议按 章节导航 index.md 的顺序阅读。【免费下载链接】hello-algo《Hello 算法》动画图解、一键运行的数据结构与算法教程。支持简中、繁中、English、日本語提供 Python, Java, C, C, C#, JS, Go, Swift, Rust, Ruby, Kotlin, TS, Dart 等代码实现项目地址: https://gitcode.com/GitHub_Trending/he/hello-algo创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考