一、前言
在上一篇文章中,我们了解到了通过向上调整建堆,以及向下调整建堆。向下调整建堆需要满足左右子树都是堆的情况。如果左右子树不是堆?我们又该如何用向下调整来建堆呢。
二、原理
向下调整算法要求左右子树必须是堆,我们拿到一个左右子树不是堆的数组,比如:
这种情况下,我们要使用向下调整算法就不能从根结点开始,因为此时根结点的左右子树并不是堆。我们需要从第一个非叶子结点开始调整。
以第一个非叶子结点来看,它的左右子树都是叶子结点。可以看作大堆/小堆,所以它是满足左右子树都是堆的特点的。
以小根堆为例子:
1.第一个非叶子结点使用向下调整函数,与比它小的子结点交换。
2.第一个非叶子结点下标-1,得到第二个非叶子结点。再向下调整。
3.重复上述两个步骤。到了根结点的时候,左右子树已经是两个小根堆了,再使用一次向下调整的函数,就可以得到堆。
注意:
第一个非叶子结点就是最后一个叶子节点的父结点。parent = (child-1)/2;
三、函数实现
1.主函数实现
int main()
{
int arr[] = { 25,18,12,20,10,36,45 };
int sz = sizeof(arr) / sizeof(arr[0]);
HeapPrint(arr, sz);
int parent = (sz - 1 - 1) / 2;//第一个非叶子结点
while (parent >= 0)
{
AdjustDown(arr, parent, sz);
--parent;
}
HeapPrint(arr, sz);
return 0;
}
2.其他函数实现
void swap(int* child, int* parent)
{
int temp = *child;
*child = *parent;
*parent = temp;
}
void AdjustDown(int* heap,int parent,int n)
{
int child = parent * 2 + 1;
while (child < n)
{
if (child + 1 < n && heap[child] > heap[child + 1])
{
++child;
}
if (heap[child] < heap[parent])
{
swap(&heap[child], &heap[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
void HeapPrint(int* arr, int sz)
{
for (int i = 0; i < sz; i++)
{
printf("%d ", arr[i]);
}
printf("\n");
}
四、运行结果分析
上面第一排是初始的数组数据,不是堆
下面是经过向下调整以后的堆
五、总结
1.获取第一个非叶子结点的下标:parent = (child-1)/2;
2.调用向下调整函数
3.parent–;
4.重复2.3步骤,循环条件parent >=0;即根结点也进行了向下调整。这个时候一个大/小堆就生成了。
下一篇讲解向上调整函数以及向上/向下调整函数的时间复杂度分析。