Fork me on GitHub

Heapsort 堆排序算法详解（Java实现)

Heapsort (堆排序)是最经典的排序算法之一，在google或者百度中搜一下可以搜到很多非常详细的解析。同样好的排序算法还有quicksort（快速排序）和merge sort（归并排序），选择对这个算法进行分析主要是因为它用到了一个非常有意思的算法技巧：数据结构 - 堆。而且堆排其实是一个看起来复杂其实并不复杂的排序算法，个人认为heapsort在机器学习中也有重要作用。这里重新详解下关于Heapsort的方方面面，也是为了自己巩固一下这方面知识，有可能和其他的文章有不同的入手点，如有错误，还请指出。文中引用的referecne会再结尾标注。

p.s. 个人认为所谓详解是你在看相关wiki或者算法书看不懂的时候看通俗易懂的解释，不过最佳方案还是去看教授们的讲解，推荐reference[1]中的heapsort章节。

 

以上是废话，可以不看

 

 Section 1 - 简介

Heapsort是一个comparison-based的排序算法（快排，归并，插入都是；counting sort不是），也是一种选择排序算法（selection sort），一个选择算法（selection algorithm）的定义是找到一个序列的k-th order statistic（统计学中的术语），直白的说就是找到一个list中第k-th小的元素。以上都可以大不用懂，heapsort都理解了回来看一下是这回事就是了。同样，插值排序也是一种选择排序算法。

 

Heapsort的时间复杂度在worst-case是\(O(nlgn)\)，average-case是\(O(nlgn)\)；空间复杂度在worst-case是\(O(1)\)，也就是说heapsort可以in-place实现；heapsort不稳定。

 

以下顺便附上几种排序算法的时间复杂度比较（\(\Theta-notation\)比\(O-notation\)更准确的定义了渐进分析（asymptotic analysis）的上下界限，详细了解可以自行google）：

Table 1 - 四种排序算法的running time比较
Algorithm

Worst-case

Average-case/expected 

Insertion sort（插值排序）

  \(\Theta (n^2)\)  \(\Theta (n^2)\)

Merge sort（归并排序）

  \(\Theta (nlgn)\)  \(\Theta (nlgn)\)
Heapsort（堆排序）  \(O(nlgn)\)  \(O(nlgn)\)
Quicksort（快速排序）  \(\Theta (n^2)\)  \(\Theta (n^2)\) (expected)

*Additional Part - KNN

heapsort在实践中的表现经常不如quicksort（尽管quicksort最差表现为 \(\Theta (n^2)\)，但quicksort 99%情况下的runtime complexity为 \(\Theta (nlgn)\)），但heapsort的\(O(nlgn)\)的上限以及固定的空间使用经常被运作在嵌入式系统。在搜索或机器学习中经常也有重要的作用，它可以只返回k个排序需要的值而不管其他元素的值。例如KNN（K-nearest-neighbour）中只需返回K个最小值即可满足需求而并不用对全局进行排序。当然，也可以使用divide-and-conquer的思想找最大/小的K个值，这是一个题外话，以后有机会做一个专题比较下。

以下程序为一个简单的在python中调用heapq进行heapsort取得k个最小值，可以大概体现上面所述的特性：

 1 '''
 2 Created On 15-09-2014
 3 
 4 @author: Jetpie
 5 
 6 '''
 7 
 8 
 9 import heapq, time
10 import scipy.spatial.distance as spd
11 import numpy as np
12 
13 pool_size = 100000
14 
15 #generate an 3-d random array of size 10,000
16 # data = np.array([[2,3,2],[3,2,1],[2,1,3],[2,3,2]])
17 data = np.random.random_sample((pool_size,3))
18 #generate a random input
19 input = np.random.random_sample()
20 #calculate the distance list
21 dist_list =  [spd.euclidean(input,datum) for datum in data]
22 
23 #find k nearest neighbours
24 k = 10
25 
26 #use heapsort
27 start = time.time()
28 heap_sorted = heapq.nsmallest(k, range(len(dist_list)), key = lambda x: dist_list[x])
29 print('Elasped time for heapsort to return %s smallest: %s'%(k,(time.time() - start))) 
30 
31 #find k nearest neighbours
32 k = 10000
33 
34 #use heapsort
35 start = time.time()
36 heap_sorted = heapq.nsmallest(k, range(len(dist_list)), key = lambda x: dist_list[x])
37 print('Elasped time for heapsort to return %s smallest: %s'%(k,(time.time() - start)))
get_k_smallest

 运行结果为:

Elasped time for heapsort to return 10 smallest: 0.0350000858307
Elasped time for heapsort to return 10000 smallest: 0.0899999141693

 

 Section 2 - 算法过程理解

2.1 二叉堆

完全二叉树　　　binary heap

在“堆排序”中的“堆”通常指“二叉堆（binary heap)”，许多不正规的说法说“二叉堆”其实就是一个完全二叉树（complete binary tree），这个说法正确但不准确。但在这基础上理解“二叉堆”就非常的容易了，二叉堆主要满足以下两项属性（properties）：

#1 - Shape Property: 它是一个完全二叉树。

#2 - Heap Property: 主要分为max-heap property和min-heap property（这就是我以前说过的术语，很重要）

　　|--max-heap property ：对于所有除了根节点（root）的节点 i，\(A[Parent] \geq A[i]\)

　　|--min-heap property ：对于所有除了根节点（root）的节点 i，\(A[Parent] \leq A[i]\)

上图中的两个二叉树结构均是完全二叉树，但右边的才是满足max-heap property的二叉堆。

在以下的描述中，为了方便，我们还是用堆来说heapsort中用到的二叉堆。

 

2.2 一个初步的构想

有了这样一个看似简单的结构，我们可以产生以下初步构想来对数组A做排序：

1.将A构建成一个最大堆（符合max-heap property,也就是根节点最大）；

2.取出根节点（how?）；

3.将剩下的数组元素在建成一个最大二叉堆，返回第2步，直到所有元素都被取光。

 

如果已经想到了以上这些，那么就差不多把heapsort完成了，剩下的就是怎么术语以及有逻辑、程序化的表达这个算法了。

 

2.3 有逻辑、程序化的表达

通常，heapsort使用的是最大堆（max-heap）。给一个数组A（我们使用 Java序列[0...n]），我们按顺序将它初始化成一个堆：

Input:　数组A

Initialization: 堆初始化

*堆的根节点（root）为A[0];

对这个堆中index为\(i\)的节点，我们可以得到它的parent, left child and right child，有以下操作：

Parent(i): \(parent(i)\gets A[floor((i-1)/2)]\)

Left(i): \(left(i)\gets A[2*i + 1]\)

Right(i): \(right(i)\gets A[2*i + 2]\)

通过以上操作，我们可以在任意index-\(i\)得到与其相关的其他节点（parent/child)。

在heapsort中，还有三个非常重要的基础操作（basic procedures):

Max-Heapify(A , i): 维持堆的#2 - Heap Property，别忘了在heapsort中我们指的是max-heap property（min-heap property通常是用来实现priority heap的，我们稍后提及）。

Build-Max-Heap(A): 顾名思义，构建一个最大堆（max-heap）。

Heapsort(A): 在Build-Max-Heap(A)的基础上实现我们2.2构想中得第2-3步。

其实这三个操作每一个都是后面操作的一部分。

下面我们对这三个非常关键的步骤进行详细的解释。

 

Max-Heapify(A , i)

+Max-Heapify的输入是当前的堆A和index-\(i\)，在实际的in-place实现中，往往需要一个heapsize也就是当前在堆中的元素个数。

+Max-Heapify有一个重要的假设：以Left(\(i\))和Right(\(i\))为根节点的subtree都是最大堆（如果树的知识很好这里就很好理解了，但为什么这么假设呢？在Build-Max-Heap的部分会解释）。 

+有了以上的输入以及假设，那么只要对A[i], A[Left(i)]和A[Right(i)]进行比较，那么会产生两种情况：

　　-第一种，最大值（\(largest\)）是A[i]，那么基于之前的重要假设，以\(i\)为根节点的树就已经符合#2 - Heap Property了。

　　-第二种，最大值（\(largest\)）是A[Left(i)]或A[Right(i)]，那么交换A[i]与A[\(largest\)]，这样的结果是以\(largest\)为根节点的subtree有可能打破了#2 - Heap Property，那么对以\(largest\)为根节点的树进行Max-Heapify(A, largest)的操作。

+以上所述的操作有一个形象的描述叫做A[i] “float down", 使以\(i\)为根节点的树是符合#2 - Heap Property的，以下的图例为A[0] ”float down"的过程（注意，以A[1]和A[2]为根节点的树均是最大堆）。

 heapify step 1　　heapify step2　　heapify step 3　　heapify step 4

以下附上reference[1]中的Psudocode：

 1 MAX-HEAPIFY（A, i） 
 2     l = LEFT(i)
 3     r = RIGHT(i)
 4     if <= heapsize and A[l] > A[i]
 5         largest = l
 6     else largest = i
 7     if r <= heapsize and A[r] > A[largest]
 8         largest = r
 9     if not largest = i
10         exchange A[i] with a[largest]
11         MAX-HEAPIFY（A, largest）

 

Build-Max-Heap(A)

先附上reference[1]中的Psudocode（做了部分修改，这样更明白），因为非常简单：

1 BUILD-MAX-HEAP(A)
2     heapsize = A.length
3     for i = PARENT(A.length-1) downto 0
4         MAX-HEAPIFY(A , i)

+Build-Max-Heap首先找到最后一个有子节点的节点 \(i = PARENT(A.length -1)\) 作为初始化（Initialization），因为比 i 大的其他节点都没有子节点了所以都是最大堆。

+对 i 进行降序loop并对每个 i 都进行Max-Heapify的操作。由于比 i 大的节点都进行过Max-Heapify操作而且 i 的子节点一定比 i 大， 因此符合了Max-Heapify的假设（以Left(\(i\))和Right(\(i\))为根节点的subtree都是最大堆）。

下图为对我们的输入进行Build-Max-Heap的过程：

max heap 1　　max heap 2　　max heap 3　　

max heap 4　　max heap 5　　max heap 6

 

Heapsort(A)

到现在为止我们已经完成了2.2中构想的第一步，A[0]也就是root节点是数组中的最大值。如果直接将root节点取出，会破坏堆的结构，heapsort算法使用了一种非常聪明的方法。

+将root节点A[0]和堆中最后一个叶节点（leaf）进行交换，然后取出叶节点。这样，堆中除了以A[0]为root的树破坏了#2 - Heap Property，其他subtree仍然是最大堆。只需对A[0]进行Max-Heapify的操作。

+这个过程中将root节点取出的方法也很简单，只需将\(heapsize\gets heapsize -1\)。

下面是reference[1]中的Psudocode：

1 HEAPSORT(A):
2     BUILD-MAX-HEAP(A)
3     for i = A.length downto 1
4         exchange A[0] with A[i]
5         heapsize =  heapsize -1
6         MAX-HEAPIFY(A , 0)

 

到此为止就是整个heapsort算法的流程了。注意，如果你是要闭眼睛也能写出一个堆排，最好的方法就是理解以上六个重要的操作。 

 

 Section 3 - runtime复杂度分析

这一个section，我们对heapsort算法过程中的操作进行复杂度分析。

首先一个总结：

  • Max-Heapify ~ \(O(lgn)\)
  • Build-Max-Heap ~ \(O(n)\)
  • Heapsort ~ \(O(nlgn)\)

然后我们分析一下为什么是这样的。在以下的分析中，我们所指的所有节点\(i\)都是从1开始的。

Max-Heapify 

这个不难推导，堆中任意节点 i 到叶节点的高度（height）是\(lgn\)。要专业的推导，可以参考使用master theorem。

Build-Max-Heap

在分析heapsort复杂度的时候，最有趣的就是这一步了。

如果堆的大小为\(n\)，那么堆的高度为\(\lfloor lgn\rfloor\)；

对于任意节点\(i\)，\(i\)到叶节点的高度是\(h\)，那么高度为\(h\)的的节点最多有\(\lceil n /2^{h+1}\rceil\)个，下面是一个大概的直观证明：

　　-首先，一个大小为\(n\)的堆的叶节点（leaf）个数为\(\lceil n/2\rceil\):

　　　　--还记不记得最后一个有子节点的节点parent(length - 1)是第\(\lfloor n/2\rfloor\)（注意这里不是java序号，是第几个），由此可证叶节点的个数为n - \(\lfloor n/2\rfloor\);

　　-那么如果去掉叶节点，剩下的堆的节点个数为\(n - \lceil n/2\rceil = \lfloor n/2\rfloor\)，这个新树去掉叶节点后节点个数为\(\lfloor \lfloor n/2\rfloor /2\rfloor\) ;

　　-（这需要好好想一想）以此类推，最后一个树的叶节点个数即为高度为\(h\)的节点的个数，一定小于\(\lceil (n/2)/2^h\rceil\)，也就是\(\lceil n/2^{h+1}\rceil\)。

对于任意节点\(i\)，\(i\)到叶节点的高度是\(h\)，运行Max-Heapify所需要的时间为\(O(h)\)，上面证明过。

那么Build-Max-Heap的上限时间为（参考reference[1]）：

$\sum_{h=0}^{\lfloor lgn\rfloor } \lceil \frac{n}{2^{h+1}}\rceil O(h) = O\left(n\sum_{h=0}^{\lfloor lgn\rfloor }\frac{h}{2^h}\right)$

根据以下定理：

$\sum_{k=0}^{\infty } kx^k = \frac{x}{(1-x)^2} for \quad |x| < 1$

我们用$x = \frac{1}{2}$替换求和的部分得到：

$\sum_{h=0}^{\infty } \frac{h}{2^h} = \frac{1/2}{(1-1/2)^2} = 2$

综上所述，我们可以求得：

$O\left(n\sum_{h=0}^{\lfloor lgn\rfloor }\frac{h}{2^h}\right) = O\left(n\sum_{h=0}^{\infty}\frac{h}{2^h}\right) = O(2n) = O(n)$

Heapsort

由于Build-Max-Heap复杂度为$O(n)$，有n-1次调用Max-Heapify（复杂度为$O(lgn)$），所有总的复杂度为$O(nlgn)$

 

到此为止，所有functions的运行复杂度都分析完了，下面的章节就是使用Java的实现了。

 

 Section 4 - Java Implementation

这个Section一共有两个内容，一个简单的Java实现（只有对key排序功能）和一个Priority Queue。

Parameters & Constructors:

1 protected double A[];
2 protected int heapsize;
3 
4 //constructors
5 public MaxHeap(){}
6 public MaxHeap(double A[]){
7     buildMaxHeap(A);
8 }

求parent/left child/right child:

1 protected int parent(int i) {return (i - 1) / 2;}
2 protected int left(int i) {return 2 * i + 1;}
3 protected int right(int i) {return 2 * i + 2;}

保持最大堆特性：

protected void maxHeapify(int i){
    int l = left(i);
    int r = right(i);
    int largest = i;
    if (l <= heapsize - 1 && A[l] > A[i])
        largest = l;
    if (r <= heapsize - 1 && A[r] > A[largest])
        largest = r;
    if (largest != i) {
        double temp = A[i];
        // swap
        A[i] = A[largest];
        A[largest] = temp;
        this.maxHeapify(largest);
    }
}

 构造一个“最大堆”：

1 public void buildMaxHeap(double [] A){
2     this.A = A;
3     this.heapsize = A.length;
4     
5     for (int i = parent(heapsize - 1); i >= 0; i--)
6         maxHeapify(i);        
7 }

对一个array使用heapsort:

 1 public void heapsort(double [] A){
 2     buildMaxHeap(A);
 3     
 4     int step = 1;
 5     for (int i = A.length - 1; i > 0; i--) {
 6         double temp = A[i];
 7         A[i] = A[0];
 8         A[0] = temp;
 9         heapsize--;
10         System.out.println("Step: " + (step++) + Arrays.toString(A));
11         maxHeapify(0);
12     }        
13 }

main函数:

1 public static void main(String[] args) {
2     //a sample input
3     double [] A = {3, 7, 2, 11, 3, 4, 9, 2, 18, 0};
4     System.out.println("Input: " + Arrays.toString(A));
5     MaxHeap maxhp = new MaxHeap();
6     maxhp.heapsort(A);
7     System.out.println("Output: " + Arrays.toString(A));
8 
9 }

运行结果:

Input: [3.0, 7.0, 2.0, 11.0, 3.0, 4.0, 9.0, 2.0, 18.0, 0.0]
Step: 1[0.0, 11.0, 9.0, 7.0, 3.0, 4.0, 2.0, 2.0, 3.0, 18.0]
Step: 2[0.0, 7.0, 9.0, 3.0, 3.0, 4.0, 2.0, 2.0, 11.0, 18.0]
Step: 3[2.0, 7.0, 4.0, 3.0, 3.0, 0.0, 2.0, 9.0, 11.0, 18.0]
Step: 4[2.0, 3.0, 4.0, 2.0, 3.0, 0.0, 7.0, 9.0, 11.0, 18.0]
Step: 5[0.0, 3.0, 2.0, 2.0, 3.0, 4.0, 7.0, 9.0, 11.0, 18.0]
Step: 6[0.0, 3.0, 2.0, 2.0, 3.0, 4.0, 7.0, 9.0, 11.0, 18.0]
Step: 7[0.0, 2.0, 2.0, 3.0, 3.0, 4.0, 7.0, 9.0, 11.0, 18.0]
Step: 8[2.0, 0.0, 2.0, 3.0, 3.0, 4.0, 7.0, 9.0, 11.0, 18.0]
Step: 9[0.0, 2.0, 2.0, 3.0, 3.0, 4.0, 7.0, 9.0, 11.0, 18.0]
Step: 10[0.0, 2.0, 2.0, 3.0, 3.0, 4.0, 7.0, 9.0, 11.0, 18.0]
Output: [0.0, 2.0, 2.0, 3.0, 3.0, 4.0, 7.0, 9.0, 11.0, 18.0]

 

heapsort在实践中经常被一个实现的很好的快排打败，但heap有另外一个重要的应用，就是Priority Queue。这篇文章只做拓展内容提及，简单得说，一个priority queue就是一组带key的element，通过key来构造堆结构。通常，priority queue使用的是min-heap，例如按时间顺序处理某些应用中的objects。

为了方便，我用Inheritance实现一个priority queue：

 1 package heapsort;
 2 
 3 import java.util.Arrays;
 4 
 5 public class PriorityQueue extends MaxHeap{
 6 
 7     public PriorityQueue(){super();}
 8     public PriorityQueue(double [] A){super(A);}
 9     
10     public double maximum(){
11         return A[0];
12     }
13     
14     public double extractMax(){
15         if(heapsize<1)
16             System.err.println("no element in the heap");
17         double max = A[0];
18         A[0] = A[heapsize-1];
19         heapsize--;
20         this.maxHeapify(0);
21         return max;
22     }
23     
24     public void increaseKey(int i,double key){
25         if(key < A[i])
26             System.err.println("new key should be greater than old one");
27         
28         A[i] = key;
29         while(i>0 && A[parent(i)] <A[i]){
30             double temp = A[i];
31             A[i] = A[parent(i)];
32             A[parent(i)] = temp;
33             i = parent(i);
34         }
35     }
36     
37     public void insert(double key){
38         heapsize++;
39         A[heapsize - 1] = Double.MIN_VALUE;
40         increaseKey(heapsize - 1, key);
41     }
42     
43     public static void main(String[] args) {
44         //a sample input
45         double [] A = {3, 7, 2, 11, 3, 4, 9, 2, 18, 0};
46         System.out.println("Input: " + Arrays.toString(A));
47         PriorityQueue pq = new PriorityQueue();
48         pq.buildMaxHeap(A);
49         System.out.println("Output: " + Arrays.toString(A));
50         pq.increaseKey(2, 100);
51         System.out.println("Output: " + Arrays.toString(A));
52         System.out.println("maximum extracted: " + pq.extractMax());
53         pq.insert(33);
54         System.out.println("Output: " + Arrays.toString(A));    
55 
56     }
57 }
priorityqueue

 运行结果：

Input: [3.0, 7.0, 2.0, 11.0, 3.0, 4.0, 9.0, 2.0, 18.0, 0.0]
Output: [18.0, 11.0, 9.0, 7.0, 3.0, 4.0, 2.0, 2.0, 3.0, 0.0]
Output: [100.0, 11.0, 18.0, 7.0, 3.0, 4.0, 2.0, 2.0, 3.0, 0.0]
maximum extracted: 100.0
Output: [33.0, 18.0, 4.0, 7.0, 11.0, 0.0, 2.0, 2.0, 3.0, 3.0]

 

 

Section 5 - 小结

首先要说本文全部是原创，如需要使用，只需要引用一下并不需要通知我。

写到最后发现有很多写得很冗余，也有啰嗦的地方。感觉表达出来对自己的知识巩固很有帮助。Heapsort真是一个非常有意思的排序方法，是一个通用而不算复杂的算法，这是决定开始写blogs后的第一篇文章，一定有很多不足，欢迎讨论！之后打算写一些机器学习和计算机视觉方面的来抛砖引玉。希望通过博客园这个平台可以交到更多有钻研精神的朋友。

 

Bibliography

[1] Thomas H. Cormen, Charles E. Leiserson, Ronald L. Rivest, and Clifford Stein.
Introduction to Algorithms. MIT Press, 3rd edition, 2009.

[2] Wikipedia. Heapsort — wikipedia, the free encyclopedia, 2014. [Online; accessed
15-September-2014].

 

posted @ 2014-09-16 22:25  JetPie  阅读(27347)  评论(5)  编辑  收藏  举报