forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMain.js
44 lines (40 loc) · 928 Bytes
/
Main.js
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
function swap(arr, i, j) {
const temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
function heapify(arr, n, i) {
if (i >= n) return;
const c1 = 2 * i + 1;
const c2 = 2 * i + 2;
let largest = i;
if (c1 < n && arr[c1] > arr[largest]) {
largest = c1;
}
if (c2 < n && arr[c2] > arr[largest]) {
largest = c2;
}
if (largest !== i) {
swap(arr, i, largest);
heapify(arr, n, largest);
}
}
function buildHeap(arr, n) {
const lastNode = n - 1;
const lastParent = Math.floor((lastNode - 1) / 2);
for (let i = lastParent; i >= 0; i--) {
heapify(arr, n, i);
}
}
function heapSort(arr, n) {
buildHeap(arr, n);
for (let i = n - 1; i >= 0; i--) {
swap(arr, i, 0);
heapify(arr, i, 0);
}
}
function main(arr, n, m) {
heapSort(arr, n);
const list = arr.slice(0, m);
console.log(list.join(' '));
}