概述
堆排序是一種樹形選擇排序,是對直接選擇排序的有效改進。
堆的定義如下:具有n個元素的序列(k1,k2,...,kn), 當且僅當滿足:
時稱之為堆。由堆的定義可以看出,堆頂元素(即第一個元素)必為最小項(小頂堆)或最大項(大頂堆)。
若以一維數組存儲一個堆,則堆對應一棵完全二叉樹,且所有非葉結點(有子女的結點)的值均不大于(或不小于)其子女的值,根結點(堆頂元素)的值是最小(或最大)的。
(a)大頂堆序列:(96, 83, 27, 38, 11, 09)
(b)小頂堆序列:(12, 36, 24, 85, 47, 30, 53, 91)
初始時把要排序的n 個數的序列看作是一棵順序存儲的二叉樹(一維數組存儲二叉樹),調整它們的存儲序,使之成為一個堆,將堆頂元素輸出,得到n 個元素中最小(或最大)的元素。然后對剩下的n-1個元素重新調整使之成為堆,輸出堆頂元素,得到n 個元素中次小(或次大)的元素。依此類推,直到最后得到有n個節點的有序序列。稱這個過程為堆排序。
步驟&實例
實現堆排序需解決兩個問題:
(1)如何將n 個待排序的數建成堆;
(2)輸出堆頂元素后,怎樣調整剩余n-1 個元素,使其成為一個新堆。
建堆方法(小頂堆):
對初始序列建堆的過程,就是一個反復進行篩選的過程。
n 個結點的完全二叉樹,則最后一個結點是第n/2個結點的子樹。
篩選從第n/2個結點為根的子樹開始(n/2是最后一個有子樹的結點),使該子樹成為堆。
之后向前依次對各結點為根的子樹進行篩選,使之成為堆,直到根結點。
如圖建堆初始過程
無序序列:(49, 38, 65, 97, 76, 13, 27, 49)
(a) 無序序列,初始二叉樹,97(第8/2=4個結點)為最后一個結點(49)的父結點。
(b) 97>=49,替換位置,接下來對n/2的上一個結點65進行篩選。
(c) 13<=27且65>=13,替換65和13的位置,接下來對38進行替換(都大于它,不需操作),對49進行篩選。
(d) 13<=38且49>=13,替換49和13的位置,49>=27,替換49和27的位置。
(e) 最終得到一個堆,13是我們得到的最小數。
調整堆的方法(小頂堆):
設有m 個元素的堆,輸出堆頂元素后,剩下m-1 個元素。將堆底元素送入堆頂,堆被破壞,其原因僅是根結點不滿足堆的性質。
將根結點與左、右子樹中較小元素的進行交換。
若與左子樹交換:如果左子樹堆被破壞,則重復方法(2).
若與右子樹交換,如果右子樹堆被破壞,則重復方法(2).
繼續對不滿足堆性質的子樹進行上述交換操作,直到葉子結點,堆被建成。
調整堆只需考慮被破壞的結點,其他的結點不需調整。
代碼實現(Java)
運行代碼結合注釋與上面的實例步驟進行對比思考。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
package com.coder4j.main; public class HeapSort { /** * 調整為小頂堆(排序后結果為從大到小) * * @param array是待調整的堆數組 * @param s是待調整的數組元素的位置 * @param length是數組的長度 * */ public static void heapAdjustS( int [] array, int s, int length) { int tmp = array[s]; int child = 2 * s + 1 ; // 左孩子結點的位置 System.out.println( "待調整結點為:array[" + s + "] = " + tmp); while (child < length) { // child + 1 是當前調整結點的右孩子 // 如果有右孩子且小于左孩子,使用右孩子與結點進行比較,否則使用左孩子 if (child + 1 < length && array[child] > array[child + 1 ]) { child++; } System.out.println( "將與子孩子 array[" + child + "] = " + array[child] + " 進行比較" ); // 如果較小的子孩子比此結點小 if (array[s] > array[child]) { System.out.println( "子孩子比其小,交換位置" ); array[s] = array[child]; // 把較小的子孩子向上移動,替換當前待調整結點 s = child; // 待調整結點移動到較小子孩子原來的位置 array[child] = tmp; child = 2 * s + 1 ; // 繼續判斷待調整結點是否需要繼續調整 if (child >= length) { System.out.println( "沒有子孩子了,調整結束" ); } else { System.out.println( "繼續與新的子孩子進行比較" ); } // continue; } else { System.out.println( "子孩子均比其大,調整結束" ); break ; // 當前待調整結點小于它的左右孩子,不需調整,直接退出 } } } /** * 調整為大頂堆(排序后結果為從小到大) * * @param array是待調整的堆數組 * @param s是待調整的數組元素的位置 * @param length是數組的長度 * */ public static void heapAdjustB( int [] array, int s, int length) { int tmp = array[s]; int child = 2 * s + 1 ; // 左孩子結點的位置 System.out.println( "待調整結點為:array[" + s + "] = " + tmp); while (child < length) { // child + 1 是當前調整結點的右孩子 // 如果有右孩子且大于左孩子,使用右孩子與結點進行比較,否則使用左孩子 if (child + 1 < length && array[child] < array[child + 1 ]) { child++; } System.out.println( "將與子孩子 array[" + child + "] = " + array[child] + " 進行比較" ); // 如果較大的子孩子比此結點大 if (array[s] < array[child]) { System.out.println( "子孩子比其大,交換位置" ); array[s] = array[child]; // 把較大的子孩子向上移動,替換當前待調整結點 s = child; // 待調整結點移動到較大子孩子原來的位置 array[child] = tmp; child = 2 * s + 1 ; // 繼續判斷待調整結點是否需要繼續調整 if (child >= length) { System.out.println( "沒有子孩子了,調整結束" ); } else { System.out.println( "繼續與新的子孩子進行比較" ); } // continue; } else { System.out.println( "子孩子均比其小,調整結束" ); break ; // 當前待調整結點大于它的左右孩子,不需調整,直接退出 } } } /** * 堆排序算法 * * @param array * @param inverse true 為倒序排列,false 為正序排列 */ public static void heapSort( int [] array, boolean inverse) { // 初始堆 // 最后一個有孩子的結點位置 i = (length - 1) / 2, 以此向上調整各結點使其符合堆 System.out.println( "初始堆開始" ); for ( int i = (array.length - 1 ) / 2 ; i >= 0 ; i--) { if (inverse) { heapAdjustS(array, i, array.length); } else { heapAdjustB(array, i, array.length); } } System.out.println( "初始堆結束" ); for ( int i = array.length - 1 ; i > 0 ; i--) { // 交換堆頂元素H[0]和堆中最后一個元素 int tmp = array[i]; array[i] = array[ 0 ]; array[ 0 ] = tmp; // 每次交換堆頂元素和堆中最后一個元素之后,都要對堆進行調整 if (inverse) { heapAdjustS(array, 0 , i); } else { heapAdjustB(array, 0 , i); } } } public static void main(String[] args) { int [] array = { 49 , 38 , 65 , 97 , 76 , 13 , 27 , 49 }; heapSort(array, false ); for ( int i : array) { System.out.print(i + " " ); } } } |
運行結果:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
初始堆開始 待調整結點為:array[3] = 97 將與子孩子 array[7] = 49 進行比較 子孩子比其小,交換位置 沒有子孩子了,調整結束 待調整結點為:array[2] = 65 將與子孩子 array[5] = 13 進行比較 子孩子比其小,交換位置 沒有子孩子了,調整結束 待調整結點為:array[1] = 38 將與子孩子 array[3] = 49 進行比較 子孩子均比其大,調整結束 待調整結點為:array[0] = 49 將與子孩子 array[2] = 13 進行比較 子孩子比其小,交換位置 繼續與新的子孩子進行比較 將與子孩子 array[6] = 27 進行比較 子孩子比其小,交換位置 沒有子孩子了,調整結束 初始堆結束 待調整結點為:array[0] = 97 將與子孩子 array[2] = 27 進行比較 子孩子比其小,交換位置 繼續與新的子孩子進行比較 將與子孩子 array[6] = 49 進行比較 子孩子比其小,交換位置 沒有子孩子了,調整結束 待調整結點為:array[0] = 97 將與子孩子 array[1] = 38 進行比較 子孩子比其小,交換位置 繼續與新的子孩子進行比較 將與子孩子 array[3] = 49 進行比較 子孩子比其小,交換位置 沒有子孩子了,調整結束 待調整結點為:array[0] = 65 將與子孩子 array[1] = 49 進行比較 子孩子比其小,交換位置 繼續與新的子孩子進行比較 將與子孩子 array[4] = 76 進行比較 子孩子均比其大,調整結束 待調整結點為:array[0] = 76 將與子孩子 array[2] = 49 進行比較 子孩子比其小,交換位置 沒有子孩子了,調整結束 待調整結點為:array[0] = 97 將與子孩子 array[1] = 65 進行比較 子孩子比其小,交換位置 沒有子孩子了,調整結束 待調整結點為:array[0] = 76 將與子孩子 array[1] = 97 進行比較 子孩子均比其大,調整結束 待調整結點為:array[0] = 97 97 76 65 49 49 38 27 13 |
PS:堆排序與直接插入排序的區別
直接選擇排序中,為了從R[1..n]中選出關鍵字最小的記錄,必須進行n-1次比較,然后在R[2..n]中選出關鍵字最小的記錄,又需要做n-2次比較。事實上,后面的n-2次比較中,有許多比較可能在前面的n-1次比較中已經做過,但由于前一趟排序時未保留這些比較結果,所以后一趟排序時又重復執行了這些比較操作。
堆排序可通過樹形結構保存部分比較結果,可減少比較次數。