首先明确维护最大堆的前提条件是:根A[i]的左右子树LEFT[i]和RIGHT[i]都是最大堆,但根A[i]可能小于左右孩子。
1.递归版
主要思想:程序的每一步是找到A[i]和A[LEFT[i]]和A[RIGHT[i]]的最大值,然后跟A[i]交换,但交换后下标是largest的结点的值可能是原来的A[i]的值。这就可能破坏最大堆性质,需要该子树进行递归调用MAX-HEAPIFY。
伪代码如下:
其中,n为最大堆的大小。
2.非递归版
主要思想:由上递归可知,我们只需要把每一次递归中的largest这个值成为新的i值即可。
代码如下:
void MAXHEAPIFY(int *a,int i,int heapSize)
{
int left=i;//左子树根节点索引
int right=i;//右子树根节点索引
int tmp;
int largest=i;
while(i<heapSize)
{
left=i<<1;//左子索引
right=(i<<1)+1;//(移位运算比算数运算优先级低,切记!)右子索引
largest=i;
if(left<=heapSize&&a[i]<a[left])
{
largest=left;
}
if(right<=heapSize&&a[largest]<a[right])
{
largest=right;
}
if(i!=largest)//不加这个判断可能会产生死循环,一个劲的在i这打转
{
tmp=a[i];
a[i]=a[largest];
a[largest]=tmp;
i=largest;//其实消除递归就要看每次递归变化的那个量,这里是关键点
}
else
{
break;
}
}
}