二叉堆的概念:是一种数组对象,可以被视为一棵完全二叉树,树的每一层都是填满的,最后一层可能除外。
二叉树有两种:最大堆和最小堆。最大堆:父节点不小于子节点。最小堆:父节点不大于子节点。在堆排序中我们使用最大堆;最小堆通常在构造优先队列时使用。
进行堆排序分为三个模块:1.保持最大堆性质;2.建堆;3:进行排序。
1.保持最大堆性质即使以i为根的子树成为最大堆
可以由下图为例:
具体程序如下:
/*****************************************\
输入:原始数组arrayA 父节点的下标i
功能:使以i为根的子树成为最大堆
时间复杂度:lgn即树的层数
\*****************************************/
void MaxHeapify(int* arrayA,int n,int i)//i为父节点的在数组的下标
{
int Length=n;
int l=2*i;//l为左子节点的在数组的下标
int r=l+1;//r为右子节点的在数组的下标
int largest=0;
int temp=0;
if((l<Length)&&(arrayA[l]>arrayA[i]))
largest=l;
else
largest=i;
if((r<Length)&&(arrayA[r]>arrayA[largest]))
largest=r;
if(largest!=i)
{
temp=arrayA[i];
arrayA[i]=arrayA[largest];
arrayA[largest]=temp;
MaxHeapify(arrayA,n,largest);
}
}
2.建堆:使数组arrayA中的元素成为最大堆
具体程序如下:
/*****************************************\
输入:原始数组arrayA
功能:使数组arrayA中的元素成为最大堆
时间复杂度:nlgn
\*****************************************/
void BuildMaxHeap(int* arrayA,int n)
{
for(int i=n/2;i>0;i--)
MaxHeapify(arrayA,n,i);
}
3.堆排序
主要思想是将每次的堆的顶节点与最末的叶节点进行交换,然后重新根据最大堆性质使得顶节点(根)成为最大值,如此循环。
具体程序如下:
/*****************************************\
输入:原始数组arrayA
功能:进行从小到大的排序
时间复杂度:nlgn
\*****************************************/
void HeapSort(int* arrayA,int n)
{
int temp=0;
int Length=n;
for(int i=Length-1;i>=2;i--)
{
temp=arrayA[1];
arrayA[1]=arrayA[i];
arrayA[i]=temp;
n--;
MaxHeapify(arrayA,n,1);
}
}
下面将三个步骤综合起来,总的排序算法程序如下:
#include<iostream>
#include<ctime>
using namespace std;
void MaxHeapify(int* arrayA,int n,int i);//保持最大堆的性质
void BuildMaxHeap(int* arrayA,int n);//构造堆
void HeapSort(int* arrayA,int n);//进行堆排序
void main()
{
int arrayA[11]={0,4,1,3,2,16,9,10,14,8,7};//第一个空间不用,是为了方便下标计算
int Length=sizeof(arrayA)/sizeof(int);//数组的长度
BuildMaxHeap(arrayA,Length);//利用数组arrayA建立最大堆
cout<<"原序列为:";
for(int i=0;i<Length;i++)
cout<<arrayA[i]<<" ";
cout<<endl;
HeapSort(arrayA,Length);
cout<<"排序好的序列为:";
for(int i=0;i<Length;i++)
cout<<arrayA[i]<<" ";
cout<<endl;
}
/*****************************************\
输入:原始数组arrayA 父节点的下标i
功能:使以i为根的子树成为最大堆
时间复杂度:lgn即树的层数
\*****************************************/
void MaxHeapify(int* arrayA,int n,int i)//i为父节点的在数组的下标
{
int Length=n;
int l=2*i;//l为左子节点的在数组的下标
int r=l+1;//r为右子节点的在数组的下标
int largest=0;
int temp=0;
if((l<Length)&&(arrayA[l]>arrayA[i]))
largest=l;
else
largest=i;
if((r<Length)&&(arrayA[r]>arrayA[largest]))
largest=r;
if(largest!=i)
{
temp=arrayA[i];
arrayA[i]=arrayA[largest];
arrayA[largest]=temp;
MaxHeapify(arrayA,n,largest);
}
}
/*****************************************\
输入:原始数组arrayA
功能:使数组arrayA中的元素成为最大堆
时间复杂度:nlgn
\*****************************************/
void BuildMaxHeap(int* arrayA,int n)
{
for(int i=n/2;i>0;i--)
MaxHeapify(arrayA,n,i);
}
/*****************************************\
输入:原始数组arrayA
功能:进行从小到大的排序
时间复杂度:nlgn
\*****************************************/
void HeapSort(int* arrayA,int n)
{
int temp=0;
int Length=n;
for(int i=Length-1;i>=2;i--)
{
temp=arrayA[1];
arrayA[1]=arrayA[i];
arrayA[i]=temp;
n--;
MaxHeapify(arrayA,n,1);
}
}
注意:我是在vs2008上运行的,与vc 6.0有点区别,主要是循环体中的循环变量的作用域,出错体现在循环变量的重复定义上。例如:在vs2008或vs2010上,程序为:
#include<stdio.h>
void main()
{
int i=0;
for(int i=0;i<5;i++)
printf("%d ",i);
}
则在VC 6.0上需改为:
#include<stdio.h>
void main()
{
int i=0;
for(i=0;i<5;i++)
printf("%d ",i);
}
原文:http://blog.csdn.net/tengweitw/article/details/9152899
作者:nineheadedbird