堆是完全二叉树的结构,因此对于一个有n个节点的堆,高度为O(logn)。
最大堆:堆中的最大元素存放在根节点的位置。 除了根节点,其他每个节点的值最多与其父节点的值一样大。也就是任意一个子树中包含的所有节点的值都不大于树根节点的值。
堆中节点的位置编号都是确定的,根节点编号为1,每一层从左到右依次编号。由堆是完全二叉树,可以知道当堆中某个节点的编号为i时,如果这个节点有左右子树,那么左子树的节点编号为2*i,右子树的节点编号为2*i+1(当然这是在根节点编号为1的情况时)。
如下图所示:
堆排序算法:
初始时候算法是利用buildMaxHeap将数组A[1...n]建成最大堆,因为数组中最大元素总在根节点A[1]处,通过把他与A[n]进行互换,我们可以让该元素放到正确位置。这时候,如果我们从堆中去掉结点n,剩余结点仍然是最大堆,而新的结点可能会违背最大堆的性质。为了维护最大堆的性质,我们调用maxHeap(A, 1),从而在A[1...n-1]上构造一个新的最大堆。堆排序要做的是不断重重复这样的操作,直到堆的大小从n-1降到2.
伪代码及图解如下:
HEAPSORT(A)
1:buildMaxHeap(A)
2: for i=A.length downto 2
3: exchange A[1] with A[I]
4: maxHeap(A, 1)
通过上面观察可知,最大堆的排序过程其实是和最大堆的删除操作类似,由于最大堆的删除只能在根结点进行,当将根结点删除完成之后,就是将剩下的结点进行整理让其符合最大堆的标准。
代码实现如下:
#include<iostream>
using namespace std;
void heapSort(int a[], int n);
void maxHeap(int a[], int n);
void buildMaxHeap(int a[], int n);
int size;
int main()
{
int a[] = { 9, 12, -3, 0, 6, 8, 15, 7 };
size= sizeof(a) / 4;
heapSort(a, size);
for (int i = 0; i < 8; i++)
cout << a[i] << " ";
return 0;
}
void heapSort(int a[], int n)
{
int i;
buildMaxHeap(a, n);
for (i = size; i>1; i--)
{
swap(a[0], a[i-1]);
size = size - 1;
maxHeap(a, 1);
}
}
void buildMaxHeap(int a[], int n)
{
int i = n / 2;
for (i; i > 0; i--)
maxHeap(a, i);
}
void maxHeap(int a[], int n)
{
int leftChild, rightChild, largest;
leftChild = 2 * n;
rightChild = 2 * n + 1;
int q = sizeof(a);
if (leftChild<=size&&a[leftChild-1]>a[n-1])
largest = leftChild;
else
largest = n;
if (rightChild<=size&&a[rightChild-1]>a[largest-1])
largest = rightChild;
if (largest != n)
{
swap(a[n-1], a[largest-1]);
maxHeap(a, largest);
}
}
结果如下: