堆——神奇的优先队列(下)
--转自啊哈磊【坐在马桶上看算法】算法12:堆——神奇的优先队列(下)
接着上一Pa说。就是如何建立这个堆呢。可以从空的堆开始,然后依次往堆中插入每一个元素,直到所有数都被插入(转移到堆中为止)。因为插入第i个元素的所用的时间是O(log i),所以插入所有元素的整体时间复杂度是O(NlogN),代码如下。
n=0; for(i=1;i<=m;i++) { n++; h[ n]=a[ i]; //或者写成scanf("%d",&h[ n]); siftup(); }
其实我们还有更快得方法来建立堆。它是这样的。
直接把99、5、36、7、22、17、46、12、2、19、25、28、1和92这14个数放入一个完全二叉树中(这里我们还是用一个一维数组来存储完全二叉树)。
在这个棵完全二叉树中,我们从最后一个结点开始依次判断以这个结点为根的子树是否符合最小堆的特性。如果所有的子树都符合最小堆的特性,那么整棵树就是最小堆了。如果这句话没有理解不要着急,继续往下看。
首先我们从叶结点开始。因为叶结点没有儿子,所以所有以叶结点为根结点的子树(其实这个子树只有一个结点)都符合最小堆的特性(即父结点的值比子结点的值小)。这些叶结点压根就没有子节点,当然符合这个特性。因此所有叶结点都不需要处理,直接跳过。从第n/2个结点(n为完全二叉树的结点总数,这里即7号结点)开始处理这棵完全二叉树。注意完全二叉树有一个性质:最后一个非叶结点是第n/2个结点。
以7号结点为根的子树不符合最小堆的特性,因此要向下调整。
同理以6号、5号和4结点为根的子树也不符合最小对的特性,都需要往下调整。
下面是已经对7号、6号、5号和4结点为根结点的子树调整完毕之后的状态。
当然目前这棵树仍然不符合最小堆的特性,我们需要继续调整以3号结点为根的子树,即将3号结点向下调整。
同理继续调整以2号结点为根的子树,最后调整以1号结点为根的子树。调整完毕之后,整棵树就符合最小堆的特性啦。
小结一下这个创建堆的算法。把n个元素建立一个堆,首先我可以将这n个结点以自顶向下、从左到右的方式从1到n编码。这样就可以把这n个结点转换成为一棵完全二叉树。紧接着从最后一个非叶结点(结点编号为n/2)开始到根结点(结点编号为1),逐个扫描所有的结点,根据需要将当前结点向下调整,直到以当前结点为根结点的子树符合堆的特性。虽然讲起来起来很复杂,但是实现起来却很简单,只有两行代码如下:
for(i=n/2;i>=1;i--) siftdown(i);
//获取堆顶的元素 int getHeapTopElement() { int t; t = heap[1];//用一个临时变量记录堆顶点的值,因为是最小堆,所以t存储的是最小值 heap[1] = heap[heap_capacity];//将堆的最后一个点赋值到堆顶 heap_capacity--;//堆的元素减少1 siftdown(1);//向下调整函数,传入需要向下调整的的结点编号,传入1即从堆的顶点开始向下调整 return t;//返回之前记录的堆的顶点的最小值 }
建立最小堆以及堆排序(从小到大)的完整代码如下:
/* Author:Mengmeng Time:2016-7-6 12:05:25 Description: 堆的使用: a.建立堆 b.使用堆来排序 */ #include <iostream> using namespace std; int heap[101];//用来存放堆的数组 int heap_capacity;//用来存储堆中元素的个数,也就是堆的大小 //交换函数,用来交换堆中的两个元素的值 void swap(int x, int y) { int t; t = heap[x]; heap[x] = heap[y]; heap[y] = t; } //向下调整建立最小堆函数 void siftDownToSetUpMinHeap(int i) //传入一个需要向下调整的结点编号i,这里传入1,即从堆的顶点开始向下调整 { int t, flag = 0;//flag用来标记是否需要继续向下调整 //当i结点有儿子的时候(其实是至少有左儿子的情况下)并且有需要继续调整的时候循环窒执行 while (i * 2 <= heap_capacity && flag == 0) { //首先判断他和他左儿子的关系,并用t记录值较小的结点编号 //若比他儿子大,则需向下调整,t取较小的结点(t=i*2) if (heap[i] > heap[i * 2]) t = i * 2; else t = i; //如果他有右儿子的情况下,再对右儿子进行讨论 if (i * 2 + 1 <= heap_capacity) { //如果右儿子的值更小,更新t为较小的结点编号 if (heap[t] > heap[i * 2 + 1]) t = i * 2 + 1; } //如果发现最小的结点编号不是自己,说明子结点中有比父结点更小的 if (t != i) { swap(t, i);//交换它们 i = t;//更新i为刚才与它交换的儿子结点的编号,便于接下来继续向下调整 } else flag = 1;//否则说明当前的父结点已经比两个子结点都要小了,不需要在进行调整了 } } //建立堆的函数 void creat() { int i; //从最后一个非叶结点到第1个结点依次进行向下调整 for (i = heap_capacity / 2; i >= 1; i--) { siftDownToSetUpMinHeap(i); } } //获取堆顶的元素 int getHeapTopElement() { int t; t = heap[1];//用一个临时变量记录堆顶点的值,因为是最小堆,所以t存储的是最小值 heap[1] = heap[heap_capacity];//将堆的最后一个点赋值到堆顶 heap_capacity--;//堆的元素减少1 siftDownToSetUpMinHeap(1);//向下调整函数,传入需要向下调整的的结点编号,传入1即从堆的顶点开始向下调整 return t;//返回之前记录的堆的顶点的最小值 } int main() { int i, num; //读入数的个数 cout << "读入数的个数:" << endl; cin >> num; cout << "读入" << num << "个数,空格作为分隔符:" << endl; for (i = 1; i <= num; i++) cin >> heap[i]; heap_capacity = num; //建堆 cout << "建立最小堆..." << endl; creat(); for (i = 1; i <= num; i++) cout << heap[i] << " "; cout << endl; cout << num << "个数从小到大排序为:" << endl; //调用getHeapTopElement()函数,连续调用heap_capacity次,其实也就是从小到大把数输出来 for (i = 1; i <= num; i++) cout << getHeapTopElement() << " "; cout<<endl; return 0; }
运行结果:
当然堆排序还有一种更好的方法。从小到大排序的时候不建立最小堆而建立最大堆。最大堆建立好后,最大的元素在h[ 1]。因为我们的需求是从小到大排序,希望最大的放在最后。因此我们将h[ 1]和h[ n]交换,此时h[ n]就是数组中的最大的元素。OK现在最大的元素已经归位,需要将堆的大小减1即n--,请注意,交换后还需将h[ 1]向下调整以保持堆的特性。然后再继续将h[ 1]和h[ n]交换,并将h[ 1]向下调整。如此反复,直到堆的大小变成1为止。此时数组h中的数就已经是排序好的了。
代码如下:
//堆排序 void heapsort() { while (heap_capacity >1) { swap(1, heap_capacity ); heap_capacity --; siftDownToSetUpMaxHeap(1); } }
完整的堆排序的代码如下,注意使用这种方法来进行从小到大排序需要建立最大堆。
#include <iostream> using namespace std; int heap[101];//用来存放堆的数组 int heap_capacity ;//用来存储堆中元素的个数,也就是堆的大小 //交换函数,用来交换堆中的两个元素的值 void swap(int x, int y) { int t; t = heap[x]; heap[x] = heap[y]; heap[y] = t; } //向下调整建立最大堆函数 void siftDownToSetUpMaxHeap(int i) //传入一个需要向下调整的结点编号i,这里传入1,即从堆的顶点开始向下调整 { int t, flag = 0;//flag用来标记是否需要继续向下调整 //当i结点有儿子的时候(其实是至少有左儿子的情况下)并且有需要继续调整的时候循环执行 while (i * 2 <= heap_capacity && flag == 0) { //首先判断他和他左儿子的关系,并用t记录值较大的结点编号 //若比他儿子小,则需向下调整,t取较大的结点(t=i*2) if (heap[i] < heap[i * 2]) t = i * 2; else t = i; //如果他有右儿子的情况下,再对右儿子进行讨论 if (i * 2 + 1 <= heap_capacity ) { //如果右儿子的值更大,更新t为较大的结点编号 if (heap[t] < heap[i * 2 + 1]) t = i * 2 + 1; } //如果发现最大的结点编号不是自己,说明子结点中有比父结点更大的 if (t != i) { swap(t, i);//交换它们 i = t;//更新i为刚才与它交换的儿子结点的编号,便于接下来继续向下调整 } else flag = 1;//否则说明当前的父结点已经比两个子结点都要大了,不需要再进行调整了 } } //建立堆的函数 void creat() { int i; //从最后一个非叶结点到第1个结点依次进行向上调整 for (i = heap_capacity / 2; i >= 1; i--) { siftDownToSetUpMaxHeap(i); } } //堆排序 /* 将heap[ 1]和heap[ heap_capacity ]交换,此时heap[ heap_capacity ]就是数组中的最大的元素。 OK现在最大的元素已经归位,需要将堆的大小减1即heap_capacity --, 请注意,交换后还需将heap[ 1]向下调整以保持堆的特性。 然后再继续将heap[ 1]和heap[ heap_capacity ]交换,并将heap[ 1]向下调整。 如此反复,直到堆的大小变成1为止。此时数组heap中的数就已经是排序好的了。 */ void heapsort() { while (heap_capacity >1) { swap(1, heap_capacity ); heap_capacity --; siftDownToSetUpMaxHeap(1); } } int main() { int i, num; //读入heap_capacity 个数 cout << "读入数的个数:" << endl; cin >> num; cout << "读入" << num << "个数,空格作为分隔符:" << endl; for (i = 1; i <= num; i++) cin >> heap[i]; heap_capacity = num; //建堆 cout << "建立最大堆..." << endl; creat(); for (i = 1; i <= num; i++) cout << heap[i] << " "; cout << endl; //堆排序 heapsort(); cout << num << "个数从小到大排序为:" << endl; //输出 for (i = 1; i <= num; i++) cout << heap[i] << " ";
cout<<endl; return 0; }
运行结果: