堆排序(Heap Sort)算法学习

在程序设计相关领域，堆(Heap)的概念主要涉及到两个方面：

• 一种数据结构，逻辑上是一颗完全二叉树，存储上是一个数组对象（二叉堆）。
• 垃圾收集存储区，是软件系统可以编程的内存区域。

本文所说的堆，指的是前者。

堆排序的时间复杂度是O(nlgN)，与快速排序达到相同的时间复杂度。但是在实际应用中，我们往往采用快速排序而不是堆排序。这是因为快速排序的一个好的实现，往往比堆排序具有更好的表现。堆排序的主要用途，是在形成和处理优先级队列方面。另外，如果计算要求是类优先级队列（比如，只要返回最大或者最小元素，只有有限的插入要求等），堆同样是很适合的数据结构。

基础知识

堆一般用数组表示，比如数组A数组的长度Length(A)，堆在数组中的元素个数HeapSize(A)。一般说来，HeapSize(A) <= Length(A)，因为数组A当中可能有一些元素不在堆中。

假设节点I是数组A中下标为i的节点。

• Parent(i) : return Floor(i/2); //I的父节点下标，Floor(i)表示比i小的最大整数。
• Left(i) : return 2*i; //I的左子节点
• Right(i) : return 2*i+1; //I的右子节点

含有n个元素的堆A的高度是: Floor(lgn)。

堆的基本操作

• MaxHeapify( A, i ):

  保持堆的性质。假设数组A和下标i，假定以Left(i)和Right(i)为根结点的左右两棵子树都已经是最大堆，节点i的值可能小于其子节点。调整节点i的位置。

• BuildMaxHeap( A ):

  从一个给定的数组建立最大堆。子数组A[ floor(n/2)+1 .... ... n]中的元素都是树的叶节点(完全二叉树的基本性质)。从索引 ceiling(n/2)开始一直到1，对每一个元素都执行MaxHeapify，最终得到一个最大堆。

• 堆排序 HeapSort( A ):

  堆排序算法的基本思想是，将数组A创建为一个最大堆，然后交换堆的根(最大元素)和最后一个叶节点x，将x从堆中去掉形成新的堆A1，然后重复以上动作，直到堆中只有一个节点。

• 优先级队列算法-增加某元素的值(优先级) : HeapIncreaseKey( A, i, key )

  增加某一个元素的优先级后（元素的值），该元素应该向上移动，才能保持堆的性质。

• 优先级队列算法-插入一个元素: Insert( S, x ) 将x元素插入到优先级队列S中。

  主要思路是，将堆的最后一个叶节点之后，扩展一个为无穷小的新叶节点，然后增大它的值为x的值。

堆排序实现原理

堆排序的C语言实现

view source

print ?

01	#include <stdio.h>

02	#include <stdlib.h>

03

04	void HeapSort(int num[],int size);

05	void BuildHeap(int num[] ,int size);

06	void PercolateDown(int num[] , int index,int size);

07	void PrintHeap(const char* strMsg,int array[],int nLength);

08	void Swap(int num[] , int v, int u);

09

10	int main(int argc, char *argv[])

11

{

12	int data[13]={8,5,4,6,13,7,1,9,12,11,3,10,2};

13	HeapSort(data,13);

14

15	system("PAUSE");

16

return 0;

17

}

18

19

20	void HeapSort(int num[] ,int size)

21

{

22

int i;

23	int iLength=size;

24

25	PrintHeap("Befor Sort:",num,iLength);

26

27	BuildHeap(num,size);// 建立小顶堆

28

29	for (i = iLength - 1; i >= 1; i--) {

30	Swap(num, 0, i);// 交换

31	size--;// 每交换一次让规模减少一次

32	PercolateDown(num, 0,size);// 将新的首元素下滤操作

33	PrintHeap("Sort Heap:",num,iLength);

34

}

35

}

36

37	// 建堆方法，只需线性时间建好

38	void BuildHeap(int num[] ,int size) {

39

int i;

40	for (i = size / 2 - 1; i >= 0; i--) {// 对前一半的节点（解释为“从最后一个非叶子节点开始，将每个父节点都调整为最小堆”更合理一些）

41	PercolateDown(num, i,size);// 进行下滤操作

42	PrintHeap("Build heap:",num,size);

43

}

44

}

45

46	// 对该数进行下滤操作，直到该数比左右节点都小就停止下滤

47	void PercolateDown(int num[] , int index,int size) {

48	int min;// 设置最小指向下标

49	while (index * 2 + 1<size) {// 如果该数有左节点，则假设左节点最小

50	min = index * 2 + 1;// 获取左节点的下标

51	if (index * 2 + 2<size) {// 如果该数还有右节点

52	if (num[min] > num[index * 2 + 2]) {// 就和左节点分出最小者

53	min = index * 2 + 2;// 此时右节点更小，则更新min的指向下标

54

}

55

}

56	// 此时进行该数和最小者进行比较，

57	if (num[index] < num[min]) {// 如果index最小，

58	break;// 停止下滤操作

59

} else {

60	Swap(num, index, min);// 交换两个数，让大数往下沉

61	index = min;// 更新index的指向

62

}

63

}// while

64

}

65

66	// 给定数组交换两个数的位置

67	void Swap(int num[] , int v, int u) {

68	int temp = num[v];

69	num[v] = num[u];

70	num[u] = temp;

71

}

72

73	void PrintHeap(const char* strMsg,int array[],int nLength)

74

{

75

int i;

76	printf("%s",strMsg);

77	for(i=0;i<nLength;i++)

78

{

79	printf("%d ",array[i]);

80

}

81	printf("\n");

82

}

下面也是C语言的实现，稍微改动了下：

 

 

01	#include <stdio.h>

02	#include <stdlib.h>

03

04	void HeapSort(int num[],int size);

05	void BuildHeap(int num[] ,int size);

06	void PercolateDown(int num[] , int index,int size);

07	void PrintHeap(const char* strMsg,int array[],int nLength);

08	void Swap(int num[] , int v, int u);

09

10	int main(int argc, char *argv[])

11

{

12	/* 将数组看成完全二叉树的中序遍历结果的线性存储 */

13	int data[13]={8,5,4,6,13,7,2,9,12,11,3,10,1};

14	HeapSort(data,13);

15

16	system("PAUSE");

17

return 0;

18

}

19

20

21	void HeapSort(int num[] ,int size)

22

{

23

int i;

24	int iLength=size;

25

26	PrintHeap("Befor Sort:",num,iLength);

27

28	BuildHeap(num,size);// 建立小顶堆

29

30	for (i = iLength - 1; i >= 1; i--) {

31	Swap(num, 0, i);// 交换

32	size--;// 每交换一次让规模减少一次

33	PercolateDown(num, 0,size);// 将新的首元素下滤操作

34	PrintHeap("Sort Heap:",num,iLength);

35

}

36

}

37

38	/* 建堆方法，只需线性时间建好；

39	建堆的结果：数组的第一个元素（即树根）是所有元素中的最小值，索引小于等于size/2-1的其它元素（即其它非叶子节点）的值都是其所在子树的最小值 */

40	void BuildHeap(int num[] ,int size) {

41

int i;

42	//从最后一个非叶子节点开始，对每个非叶子节点进型最小根调整，保证每个根节点都是其子树中的最小值

43	for (i = size / 2 - 1; i >= 0; i--) {

44	PercolateDown(num, i,size);// 进行下滤操作

45	PrintHeap("Build heap:",num,size);

46

}

47

}

48

49	/* 对该数进行下滤操作，直到该数比左右节点都小就停止下滤。

50	即对某个根节点的值进行位置下降调整，使该值比其左右子节点都小；

51	若该节点是叶子节点，则无法while循环 */

52	void PercolateDown(int num[] , int index,int size) {

53	int min;// 设置最小指向下标

54	while (index * 2 + 1<size) {// 如果该数有左节点，则假设左节点最小

55	min = index * 2 + 1;// 获取左节点的下标

56	if (index * 2 + 2<size) {// 如果该数还有右节点

57	if (num[min] > num[index * 2 + 2]) {// 就和左节点分出最小者

58	min = index * 2 + 2;// 此时右节点更小，则更新min的指向下标

59

}

60

}

61	// 此时进行该数和最小者进行比较，

62	if (num[index] < num[min]) {// 如果index最小，

63	break;// 停止下滤操作

64

} else {

65	Swap(num, index, min);// 交换两个数，让大数往下沉

66	index = min;// 更新index的指向

67

}

68

}// while

69

}

70

71	// 给定数组交换两个数的位置

72	void Swap(int num[] , int v, int u) {

73	int temp = num[v];

74	num[v] = num[u];

75	num[u] = temp;

76

}

77

78	void PrintHeap(const char* strMsg,int array[],int nLength)

79

{

80

int i;

81	printf("%s",strMsg);

82	for(i=0;i<nLength;i++)

83

{

84	printf("%d ",array[i]);

85

}

86	printf("\n");

87

}