堆排序的算法分析
堆排序算法分析
什么是堆
我们这里讨论的堆是一种数据结构,而不是垃圾收集存储机制。(二叉)堆一个数组,它可以被看成一个近似的完全二叉树,即一棵树上的每一个结点对应数组中的某一个元素,除了最底层外,该树是完全填满的,而且是从左往右填充。堆具有三个重要的属性,即父结点、左孩子、右孩子。堆又分为两种,即最大堆和最小堆,最大堆是指所有的父结点都要大于子结点,故根结点的“值”最大(这里堆中存储的不一定是数值,可以是任何一种对象,只要实现了相应的equals()方法即可,下文中将以简单的int数值类型为例来讨论和编程);而最小堆就相反,因而根结点最小。它们的性质可以用下面两个数学表达式来表示:
最大堆性质:A[PARENT(i)]>=A[i]
最小堆性质:A[PARENT(i)]<=A[i]
(注意:这里的i是除了根结点以外的所有结点)
如果把堆看成一棵树,可以定义一个堆中的结点的高度就是结点到叶结点最长简单路径上边的个数,因而可以把堆的高度定义为根结点的高度。故此,我们可以得到:
一个具有n个元素的堆的高度为logn;
高度为h的堆中,元素个数最多为2^(h+1)-1个,元素个数最少为2^h个。
因此,我们可以联想到,最大堆通过将根元素提取到结尾几乎可以对应到降序排列,而最小堆通过将根元素提取到结尾几乎可以对应到升序排列。下面我们使用最大堆来讨论。
堆的元素处理
根据堆的三个重要的属性,我们可以得到父结点和子结点的获取方法,下面用C语言实现:
/* 返回父节点和子节点索引 */ //返回父结点 int parent(int index) { return (index-1)/2; } //返回左孩子 int left(int index) { return (index<<1)+1; } //返回右孩子 int right(int index) { return (index+1)<<1; }
除了获取三个属性之外,我们还应该获取堆的大小,即元素的个数,下面用C语言实现:
/* 返回堆的大小,指针操作 */ int heapSize(int *A) { int count=0; while(*(A+count)!='\0') { count++; } return count; }
堆的性质维护
由于要满足最大堆的性质,因此当添加一个新元素时,我们需要让它满足父结点大于子结点的性质,故堆的性质需要我们去维护,因而,当我们检查结点i是否满足要求时,我们应该检查结点i是否比它的孩子结点都要大,如果比它们大,则可以不用操作;反之,应该将该结点与孩子结点进行交换到新的结点上,进而再次检查该新结点是否满足要求,操作方法同上,直到该结点没有孩子结点为止。
维护堆的性质基本思路如上所述,下面用C语言实现:
/* 维护堆的性质,即把不符合要求的元素重新组织 时间复杂度为logn */ void maxHeapify(int *A,int index,int size) { int L=left(index); int R=right(index); int tmp; int largestIndex=index; if(L<size && *(A+index)<*(A+L)) { largestIndex=L; } if(R<size && *(A+largestIndex)<*(A+R)) { largestIndex=R; } if(index!=largestIndex) { tmp=*(A+index); *(A+index)=*(A+largestIndex); *(A+largestIndex)=tmp; maxHeapify(A,largestIndex); } }
如上所示,维护堆的时间复杂度为堆的高度成正比,即logn。类似的,也可以编写一个minHeapify函数来实现最小堆。
堆的建立
堆的建立即使堆的每个元素都能满足最大堆的性质的要求,因为叶子结点没有孩子结点,因而叶子结点无法调用maxHeapify函数,故每个叶子结点都可以看成一个只包含一个元素的堆。
堆的建立可以使用以下一段C语言代码来实现:
/* 对除没有子结点的结点进行堆的维护操作,进而 使得一个数组变成一个最大堆 时间复杂度为nlogn */ void buildMaxHeap(int *A) { int size=heapSize(A); //找出不是叶子结点的最后一个索引 int need=size/2-1; int i; for(i=need;i>=0;i--) { maxHeapify(A,i,size); } }
类似的,可以编写一个buildMinHeap来建立最小堆。
堆排序算法
通过上面的步骤,我们已经把一个数组建立成为一个最大堆,然而最大堆并不一定是有序堆,因为最大堆只保证了父结点大于子结点,并没有保证左孩子一定大于右孩子,因此,堆排序就是要使左孩子大于右孩子。但是,我们始终相信,根元素是最大的,因此,我们想先找最大的比较方便,把最大的元素先扔到结尾的叶子结点上,而之前的叶子结点将跑到根结点上并且要满足堆的性质进行维护即可。
据此,我们可以设计如下算法来实现:
/* 堆排序程序 通过根元素始终是最大的,因此我们倒过来找 从最大的元素依次往最小的元素去排序 */ void heapSort(int *A) { buildMaxHeap(A); int size=heapSize(A); int i; int tmp; for(i=size-1;i>=0;i--) { tmp=*A; *A=*(A+i); *(A+i)=tmp; size--; maxHeapify(A,0,size); } }
因此堆排序的时间复杂度为nlogn。
堆排序算法的意义
堆排序是一种优秀的算法,后面我们会看到尽管快速排序的算法性能要优于堆排序,但是堆这一数据结构仍然有重要的实际应用——高校的优先队列,实际问题例如如何动态保留成绩在倒数100名的学生的信息。