堆是用数组来存储的,可以被看成一个近似的完全二叉树,树上的每一个结点对应数组中的一个元素。规定树的根节点是Array[1],即从下标1开始。
这样给定一个结点i,我们可以很容易计算它的父结点、左孩子和右孩子的下标:PARENT(i)=[i/2];LEFTCHILD(i)=2i;RIGHT(i)=2i+1;
最大堆:除了根以外的所有结点i都要满足:Array[PARENT(i)]>=Array[i];
最小堆:除了根以外的所有结点i都有:Array[PARENT(i)]<=Array[i];
在调用MAX-HEAPIFY的时候,我们假定根结点为i的左右孩子都是最大堆,但Array[i]可能小于其孩子,MAX-HEAPIFY通过让Array[i]的值在最大堆中“逐级下降”,从而使得以下标i为根节点的子树重新遵循最大堆的性质。
void MAX_HEAPIFY(T array[], int i,int N)
{
int l = i * 2;
int r = i * 2 + 1;
int larget = 0;
if (l <= N&&array[l] > array[i])
{
larget = l;
}
else
{
larget = i;
}
if (r <= N&&array[r] > array[larget])
{
larget = r;
}
if (larget != i)
{
swap(array[i], array[larget]);
MAX_HEAPIFY(array, larget, N);
}
}
我们可以在线性时间内,把一个无序数组构造成为一个最大堆。
template<typename T>
void Build_Max_Heap(T array[],int N)
{
for (int i = N / 2; i >= 1; i--)
{
MAX_HEAPIFY(array, i, N);
}
}
堆排序
利用Build_Max_Heap将输入数组array[1...N]建成最大堆,数组中最大元素总在根结点array[1]中,通过把它与array[n]交换,把最大元素放在最终位置上去。接着从堆中删除结点N,在剩余结点中调用MAX_HEAPIF(array[],1)从而在array[1...n-1]上构造一个新的最大堆。
template<typename T>
void Heap_Sort(T array[], int N)
{
Build_Max_Heap(array, N);
for (int i = N; i > 1; i--)
{
swap(array[1], array[i]);
N--;
MAX_HEAPIFY(array, 1, N);
}
}