思想

创建一个大顶堆,大顶堆的堆顶一定是最大的元素。

交换第一个元素和最后一个元素,让剩余的元素继续调整为大顶堆。

从后往前以此和第一个元素交换并重新构建,排序完成。

解法

// 交换
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

复杂度

时间复杂度:O(nlogn)

空间复杂度:O(1)

稳定性

不稳定