二叉堆结构和操作详解

在许多应用中,我们需要快速执行一些操作,比如查询和提取数据中的最大值或最小值。举个例子,当我们需要排序学生的考试成绩时,我们可能要频繁地查找和提取最高或最低分。除此之外,这种需求还广泛存在于优先级调度、数据流处理中。

假定我们要解决这样一个问题:有一个集合,每次操作都可能从中添加数据,或取出最大值,应该怎么做?假如使用暴力,仅使用一个数组来维护,我们就需要经常对数据集进行一次遍历(值是 (O(n)))。尽管简单,但如果你需要重复地进行多次这类查询,效率就很低了。这时,二叉堆作为一种高效的数据结构,提供了更好的性能。它能够在 (O(log n)) 的时间复杂度内进行最大值或最小值的查询和删除操作,解决了暴力方法中遍历整个数组的问题。

什么是二叉堆?

二叉堆是一种特殊的二叉树(Binary Tree),对于数来说分为大堆和小堆:

  • 大堆 (Max Heap): 根节点的值大于其所有子节点的值,且每个子树的根节点也满足这一性质。
  • 小堆 (Min Heap): 根节点的值小于其所有子节点的值,且每个子树的根节点也满足这一性质。

这种数据结构特别适合频繁进行最大值或最小值查询的场景,他们的用途相反,但实现原理是完全一样的,对于上例的题目来说,当然就要使用大根堆。而下图是一个典型的大根堆:

          50        /            30        40     /        /     10   20   35   25 

为什么用数组构建二叉堆?

二叉堆通常用一个数组(Array)来实现,而不是链表。这是因为:

  1. 数组实现可以省去链表中用于存储指针的额外空间。
  2. 数组实现访问子节点或父节点更加高效,通过索引计算即可完成。这样一个密集完全的二叉树更适合数组法存储。

假如设置根节点是 (0) 号,节点在第 (i) 位置,那么其左子节点在第 (2i+1) 位置,右子节点在 (2i+2) 位置,父节点在 (lfloor(i-1)/2rfloor)。假如设置根节点是 (1) 号,节点在第 (i) 位置,那么其左子节点在第 (2i) 位置,右子节点在 (2i+1) 位置,父节点在 (lfloor i/2rfloor)

建堆时,惯例是直接在原数组上操作,将原数组视为二叉堆,并调整让其符合堆的性质。每次heapify将自身往下都进行比较和调整,保证这一条路径上有序。将每个数据从后往前都进行heapify,就可以使整个数组成为堆。当然,你也可以选择从下向上调整,这是一样的。

#include <vector> #include <iostream> using namespace std;  void heapify(vector<int>& heap, int n, int i) {     int largest = i;     int left = 2 * i + 1;     int right = 2 * i + 2;      if (left < n && heap[left] > heap[largest]) {         largest = left;     }      if (right < n && heap[right] > heap[largest]) {         largest = right;     }      if (largest != i) {         swap(heap[i], heap[largest]);         heapify(heap, n, largest);     } }  void buildHeap(vector<int>& heap) {     int n = heap.size();     for (int i = n / 2 - 1; i >= 0; --i) {         heapify(heap, n, i);     } } 

处理时间复杂度: (O(n))。构建过程需要对每个非叶子节点进行调整,每次调整的最大操作次数取决于堆的高度。这看似是一个 nlog n 的过程。但实际上,通过 heapify 从最后一个非叶节点开始逐层向上调整,总体时间复杂度为 (O(n))。虽然每次 heapify 操作的时间复杂度是 (O(log n)),但由于树的高度逐层递减,每层的节点数呈指数下降。因此,整体复杂度是 (O(n)),而不是 (O(n log n))

堆操作:插入(push)和删除(pop)

1. 建堆 (Build Heap)

通过从底部到顶部逐层对二叉树进行调整,我们可以快速地将无序数组变为二叉堆。上述代码展示了具体的实现方法。

2. 插入操作:Push

插入新元素时,我们需要将该元素添加到数组尾部,然后通过从他开始,不断向上调整来维护堆的性质:

void push(vector<int>& heap, int value) {     heap.push_back(value);     int i = heap.size() - 1;      while (i != 0 && heap[(i - 1) / 2] < heap[i]) {         swap(heap[i], heap[(i - 1) / 2]);         i = (i - 1) / 2;     } } 
  • 处理时间复杂度: (O(log n)),因为堆的高度为 (log n),最多需要向上调整 (log n) 次。

3. 删除操作:Pop

删除堆的根节点时,惯例上,我们保存根节点为需要返回的值后,需要将堆尾元素移动到根节点,移除最后的节点(若不使用 vector 则需要额外的 n 保存数组当前使用的长度),然后通过一次向下调整来恢复堆的性质:

int pop(vector<int>& heap) {     if (heap.empty()) return -1; // Error case      int root = heap[0];     heap[0] = heap.back();     heap.pop_back();      heapify(heap, heap.size(), 0);     return root; } 

当然,你也可以从被删节点开始向上调整,最后删除最后节点。

  • 处理时间复杂度: (O(log n))。调整操作涉及沿树向下的路径。

由于添加和删除总是在数组尾部进行,树总会保持一颗高密度的完全二叉树形态,以保证树高为 (log n) 左右。


拓展知识

1. 堆能解决的经典问题

堆在算法设计中有广泛的应用,其功能涵盖了许多经典问题:

  • 反悔贪心问题: 在某些贪心算法中,可以通过堆动态调整选择的结果。
  • Dijkstra 算法: 优化单源最短路径算法中维护优先级队列的操作。
  • Huffman 编码: 使用最小堆构建最优前缀编码树。
  • 动态中位数问题: 利用两个堆分别维护较小和较大部分的元素。
  • 多路归并 (k) 个有序链表: 利用最小堆在每次操作中快速找到最小的头元素。

2. 堆的其他变种

二叉堆只是堆的一种,其效率适中,实现简单,但是不支持合并操作。堆的基本思想被拓展成了许多其他高级变种,每种变种适用于特定的场景:

  • 斜堆(Skew Heap): 实现简单且支持高效合并。
  • 左倾堆(Leftist Heap): 通过维护偏斜度加速合并操作。
  • 二项堆(Binomial Heap): 由一系列二项树组成,适合合并大规模数据。
  • 斐波那契堆(Fibonacci Heap): 用于优化 Dijkstra 等算法中的松弛操作。
  • 配对堆(Pairing Heap): 通过简化操作提升实际效率。
  • 优先级搜索队列(Priority Search Queue): 扩展了堆对键值对的支持。

3. 堆与优先队列的关系

优先队列是指可以进行入队和根据优先级出队的队列,而堆是实现典型优先队列的核心数据结构之一。优先队列的关键操作(插入、取最大/最小值)都可以通过堆实现高效的时间复杂度,许多语言内置了堆操作:

  • C++: 使用 std::priority_queue,默认是大根堆,小根堆需自定义比较器。
    #include <queue> std::priority_queue<int> maxHeap; // 大根堆 std::priority_queue<int, std::vector<int>, std::greater<int>> minHeap; // 小根堆 
  • Python: 使用 heapq 模块,默认是小根堆。
    import heapq heap = [] heapq.heappush(heap, value)  # 插入 smallest = heapq.heappop(heap)  # 弹出最小值 
  • Java: 使用 PriorityQueue,默认是小根堆。
    PriorityQueue<Integer> minHeap = new PriorityQueue<>(); PriorityQueue<Integer> maxHeap = new PriorityQueue<>(Collections.reverseOrder()); 
  • Go: 使用 container/heap 包。
    import "container/heap" 

通过掌握这些实现方式,可以在不同的语言中快速构建适合特定场景的优先队列,实现高效的数据处理。

发表评论

相关文章