對于堆排序會涉及一些完全二叉樹知識。對于待排序列{10, 2, 11, 8, 7}，把它看成是一顆完全二叉樹，如下圖所示。
堆分為大根堆和小根堆：大根堆表示每個根節點均大于其子節點（l(i) >= l(2i) && l(i) >= l(2i + 1)），小根堆表示每個根節點均小于其子節點（l(i) <= l(2i) && l(i) <= l(2i + 1)）。（在完全二叉樹中第i個節點的左子節點為2i，其右字節點為2i + 1）
本文將以大根堆的構建作為示例進行講解。
堆排序的第一步——構建初始堆。如何構建初始堆呢？根據定義，關鍵點在于每個根節點。觀察上述待排序列的完全二叉樹，不難發現存在節點2和節點10有子節點，它們是需要關注的節點。
如何定位節點2呢？發現它是葉子節點，或者最后一個節點的父節點，根據完全二叉樹的性質可知，除根節點外任意節點的父節點的編號為⌊n / 2⌋。已知n = 5，易知節點2的編號為⌊5 / 2⌋ = ②。比較它與左右子節點的大小并調整。
最后剩下根節點10，已知節點2的編號為②，② - 1 = ①即得到根節點10的編號。比較它與左右子節點的大小并調整。
調整完畢后發現已經構成了一個“大根堆”，示例中的待排序列較為簡單，再給出一個較為復雜的待排序列，觀察其構建大根堆的過程。對于待排序列{53, 17, 78, 09, 45, 65, 87, 32}，將它看成一顆完全二叉樹。
同樣我們來看它所需要關注的節點有哪些。
根據第一個例子，我們很容易能定位節點09的編號為⌊8 / 2⌋ = ④，節點78的編號為④ - 1 = ③……，依次類推，發現了一定的規律，即需要調整的節點位置從⌊n / 2⌋開始依次遞減直到根節點①結束(⌊n / 2⌋ ~ 1)。現在開始調整。
在第四次調整結束后發現節點53不滿足大根堆的定義，其右子節點大于它，此時需要做進一步的向下調整。
注意向下調整是每次向上調整的時候都需要做的判斷是否需要向下調整，而不是在所有的向上調整結束過后再回過頭來向下調整。這樣大根堆就建立好了，此時待排序列數組情況已經發生了改變：{87, 45, 78, 32, 17, 65, 53, 09}。接下來是如何進行排序的問題。將大根堆的根節點與最后一個節點互換，并調整二叉樹使其仍然滿足大根堆。
可以看到將根節點與最后一個節點呼喚后，待排序列的最大值已經放到了數組的最后一個位置{……, 87}，此時完成了第一趟排序，但這第一趟排序還沒有結束，此時除節點87外，其余節點并不滿足大根堆的條件，所以需要對其余節點進行調整為大根堆。排序過程不再給出，java和python3的代碼實現如下。
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
|
package com.algorithm.sort.heap;import java.util.arrays;/** * 堆排序 * created by yulinfeng on 6/20/17. */public class heap { public static void main(string[] args) { int[] nums = {53, 17, 78, 09, 45, 65, 87, 32}; nums = heapsort(nums); system.out.println(arrays.tostring(nums)); } /** * 堆排序 * @param nums 待排序數組序列 * @return 排好序的數組序列 */ private static int[] heapsort(int[] nums) { for (int i = nums.length / 2 - 1; i >= 0; i--) { heapadjust(nums, i, nums.length); } for (int i = nums.length - 1; i > 0; i--) { int temp = nums[i]; nums[i] = nums[0]; nums[0] = temp; heapadjust(nums, 0, i); } return nums; } /** * 調整堆 * * @param nums 待排序序列 * @param parent 待調整根節點 * @param length 數組序列長度 */ private static void heapadjust(int[] nums, int parent, int length) { int temp = nums[parent]; int childindex = 2 * parent + 1; //完全二叉樹節點i從編號1開始的左子節點位置在2i，此處數組下標從0開始，即左子節點所在數組索引位置為：2i + 1 while (childindex < length) { if (childindex + 1 < length && nums[childindex] < nums[childindex + 1]) { childindex++; //節點有右子節點，且右子節點大于左子節點，則選取右子節點 } if (temp > nums[childindex]) { break; //如果選中節點大于其子節點，直接返回 } nums[parent] = nums[childindex]; parent = childindex; childindex = 2 * parent + 1; //繼續向下調整 } nums[parent] = temp; }} |
python3
|
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
|
#堆排序def heap_sort(nums): for i in range(int(len(nums) / 2 - 1), -1, -1): heap_adjust(nums, i, len(nums)) for i in range(len(nums) - 1, -1, -1): temp = nums[i] nums[i] = nums[0] nums[0] = temp heap_adjust(nums, 0, i) return nums#調整堆def heap_adjust(nums, parent, length): temp = nums[parent] childindex = 2 * parent + 1 while childindex < length: if childindex + 1 < length and nums[childindex] < nums[childindex + 1]: childindex += 1 if temp > nums[childindex]: break nums[parent] = nums[childindex] parent = childindex childindex = 2 * parent + 1 nums[parent] = temp nums = [53, 17, 78, 09, 45, 65, 87, 32]nums = heap_sort(nums)print(nums) |
以上這篇老生常談比較排序之堆排序就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持服務器之家。
