function heapAdjust(arr, pos, len) {
//将pos节点的数字保存
var tmp = arr[pos];
//计算pos节点左子节点位置
var child = pos * 2 + 1;
//循环,直到没有子节点
while (child < len) {
//若pos节点有右子节点且右子节点较大,则用右子节点
if (child + 1 < len && arr[child] < arr[child + 1]) {
child++;
}
//若pos节点小于较大子节点则交换并将pos定位到子节点位置
if (arr[pos] < arr[child]) {
arr[pos] = arr[child];
pos = child;
child = pos * 2 + 1;
//将最初pos节点存到当前pos所指向的位置
arr[pos] = tmp;
} else {
break;
}
}
}
function heapSort(arr) {
//从第一个有子节点的节点开始建堆
for (var i = arr.length / 2 - 1; i >= 0; i--) {
heapAdjust(arr, i, arr.length);
}
//取出堆的最大值后继续构建最大堆
for (var i = arr.length - 1; i > 0; i--) {
var tail = arr[i];
arr[i] = arr[0];
arr[0] = tail;
//对未排序的区域继续建堆
heapAdjust(arr, 0, i);
}
}
var arr = [3, 1, 5, 7, 2, 4, 9, 6, 10, 8];
heapSort(arr);
console.log(arr);
堆排序之JavaScript实现
最新推荐文章于 2024-04-18 02:02:06 发布