堆排序之JavaScript实现

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)