本文共 2712 字,大约阅读时间需要 9 分钟。
四、(1)堆排序
第一次听堆排序是在107lab听忠哥讲的,但是没讲怎么实现。那时刚看了数据结构的二叉树,还以为要通过指针建立二叉树的方式来实现,觉得挺难的。
其实堆排序实现没有想象中的那么难。
“堆”这个词最初是在堆排序中提出来的,但后来就逐渐指”废料收集储存区“,就像程序设计语言Lisp和Java中所提供的设施那样。我们这里的堆数据结构不是废料收集存储区。
堆排序的运行时间与归并排序一样为O(n lg n), 但是一种原地(in place)排序。
(二叉)堆数据结构是一种数组对象,它可以被视为一棵完全二叉树。
对于一个数组arr[ ]={16,14,10,8,7,9,3,2,4,1}的存储数据结构如下图:
在这个结构中,对于每一个根节点i ,要保证它都比他的子节点大
我们可以用一个数组A【1...length【A】】来表示这个完全二叉树结构。 其中A【1】为根节点1
首先问题是求父节点、左儿子、右儿子的坐标,通过观察我们可以用宏或者内联函数实现:
-
- inline int Parent(int i) { return i>>1; }
- inline int Left(int i) { return i<<1; }
- inline int Right(int i) { return (i<<1)|1; }
无论是《C++ primer》还是《Effective C++》,都讲过宏的缺陷,用内联函数是个更好的选择。位运算做乘除的速度更快。
至于算法演示过程在《算法导论》上讲得很详细,不再赘述。
堆排序过程需要以下三个函数:
1、Max-Heapify(A,i) : 保持堆的性质,让A【i】在最大堆中“下降”,使以i节点为根的字数成为最大树
2、Buid-Max-Heap(A) : 自底向上将数组A【1...n】变成一个最大堆
3、Heap-Sort(A): 进行堆排序
C++代码实现:(数组A是下标1开始的)
-
- #include<cstdio>
- #include<algorithm>
- using namespace std;
-
-
-
- inline int Parent(int i) { return i>>1; }
- inline int Left(int i) { return i<<1; }
- inline int Right(int i) { return (i<<1)|1; }
-
-
- inline void Swap(int &a,int &b) { if(a!=b){a^=b; b^=a; a^=b;} }
-
-
-
-
-
-
-
-
-
-
-
-
- void MaxHeap(int *A,int heap_size,int i){
-
- int l,r,max;
- l = Left(i);
- r = Right(i);
-
- if(l<=heap_size && A[l]>A[i])
- max=l;
- else
- max=i;
-
- if(r<=heap_size && A[r]>A[max])
- max=r;
-
- if(max!=i){
- Swap(A[i],A[max]);
- MaxHeap(A,heap_size,max);
- }
- }
-
-
- void BuidMaxHeap(int *A,int heap_size){
-
- for(int i=heap_size>>1; i>=1; --i){
- MaxHeap(A,heap_size,i);
- }
- }
-
-
- void HeapSort(int *A,int heap_size){
-
- BuidMaxHeap(A,heap_size);
- int len=heap_size;
- for(int i=heap_size; i>=2; --i){
- Swap(A[1], A[i]);
- --len;
- MaxHeap(A,len,1);
- }
- }
(2)优先级队列
C++ STL中的priority_queue就是用这种方法来实现的。
1、Heap-MaxiNum(A): 取出堆中的最大值
2、Heap-Extract-Max(A): 删除堆中的最大值并返回它的值
3、Heap-Increase-Key(A,i,key):将节点i的值增加到key,这里key要比i节点原来的数大。
4、Max-Heap-Insert(A, key): 插入一个新元素key到堆中,要用到3的函数
C++实现:
-
-
- int HeapMaxNum(int *A){
- return A[1];
- }
-
- int HeapExtractMax(int *A,int heap_size){
- if(heap_size>=1){
- int max=A[1];
- A[1] = A[heap_size];
- --heap_size;
- MaxHeap(A,heap_size,1);
- return max;
- }
- }
-
- bool HeapIncreaseKey(int *A,int i,int key){
- if(key<A[i])
- return false;
- A[i] = key;
- while(i>1 && A[Parent(i)]<A[i]){
- Swap(A[i],A[Parent(i)]);
- i = Parent(i);
- }
- return true;
- }
-
- void MaxHeapInsert(int *A,int key,int heap_size){
- ++heap_size;
- A[heap_size] = -2147484646;
- HeapIncreaseKey(A,heap_size,key);
- }
转载地址:http://juzni.baihongyu.com/