思想
创建一个大顶堆,大顶堆的堆顶一定是最大的元素。
交换第一个元素和最后一个元素,让剩余的元素继续调整为大顶堆。
从后往前以此和第一个元素交换并重新构建,排序完成。
解法
// 交换
function swap(arr, a, b) {
if (a == b) { return; }
var c = arr[a];
arr[a] = arr[b];
arr[b] = c;
}
// 堆排序
function heapSort(arr) {
let n = arr.length;
if(n <= 1){
return arr;
}
buildHeap(arr);
// 堆排序
for(var i = n - 1 ; i > 0; i -- ){
swap(arr, 0, i);
maxHeapify(arr, 0, i)
}
return arr;
}
// 构建堆
function buildHeap(arr){
var n = arr.length;
var lastNode = n - 1;
var parent = Math.floor((lastNode - 1) / 2);
for(var i = parent; i >= 0; i --){
maxHeapify(arr, i, n);
}
return arr
}
// 最大堆调整
function maxHeapify(arr, i, n) {
if( i > n){
return;
}
var left = 2 * i + 1;
var right = 2 * i + 2;
var max = i;
if(arr[left] > arr[max] && left < n){
max = left;
}
if(arr[right] > arr[max] && right < n){
max = right;
}
if(max !== i){
swap(arr,i,max);
maxHeapify(arr, max, n);
}
}
console.log('heapSort([5, 2, 4, 6, 1, 3]); ',heapSort([5, 2, 4, 6, 1, 3]))
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
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
复杂度
时间复杂度:O(nlogn)
空间复杂度:O(1)
稳定性
不稳定
← 快速排序