数据结构与算法之数据结构 堆初始化 自底向上建堆 算法

数据结构与算法阿木 发布于 2025-07-11 11 次阅读


摘要:

堆是一种特殊的树形数据结构,它具有完全二叉树的性质,并且满足堆的性质:在最大堆中,父节点的值总是大于或等于其子节点的值;在最小堆中,父节点的值总是小于或等于其子节点的值。堆初始化(自底向上建堆)算法是构建堆的一种常用方法,本文将详细解析堆初始化算法的原理,并给出相应的代码实现。

一、堆初始化算法原理

堆初始化算法的基本思想是从完全二叉树的最后一个非叶子节点开始,逐个向上调整,使得每个节点都满足堆的性质。具体步骤如下:

1. 从最后一个非叶子节点开始,即最后一个父节点,其索引为 n/2 - 1,其中 n 是完全二叉树中节点的总数。

2. 从该节点开始,将其与左右子节点进行比较,如果节点值小于其左子节点值,则交换节点值与左子节点值;如果节点值小于其右子节点值,则交换节点值与右子节点值。

3. 交换后,继续向上调整,直到该节点满足堆的性质或者成为根节点。

4. 重复步骤2和3,直到所有非叶子节点都调整完毕。

二、代码实现

以下是用Python语言实现的堆初始化算法代码:

python

def heapify(arr, n, i):


largest = i


left = 2 i + 1


right = 2 i + 2

if left < n and arr[i] < arr[left]:


largest = left

if right < n and arr[largest] < arr[right]:


largest = right

if largest != i:


arr[i], arr[largest] = arr[largest], arr[i]


heapify(arr, n, largest)

def build_max_heap(arr):


n = len(arr)

从最后一个非叶子节点开始,逐个向上调整


for i in range(n // 2 - 1, -1, -1):


heapify(arr, n, i)

示例


arr = [3, 1, 6, 5, 2, 4, 8]


build_max_heap(arr)


print("初始化后的最大堆:", arr)


三、算法分析

1. 时间复杂度:堆初始化算法的时间复杂度为 O(n),其中 n 是完全二叉树中节点的总数。这是因为算法需要遍历所有非叶子节点,并且对每个节点进行一次调整。

2. 空间复杂度:堆初始化算法的空间复杂度为 O(1),因为它不需要额外的存储空间。

四、总结

堆初始化(自底向上建堆)算法是一种简单有效的构建堆的方法。通过调整完全二叉树中的节点,使得每个节点都满足堆的性质,从而得到一个堆。本文详细解析了堆初始化算法的原理,并给出了相应的代码实现。在实际应用中,堆初始化算法可以用于解决许多问题,如优先队列、选择排序等。