Js实现堆排序

Js实现堆排序

//维护堆性质 arr->数组,n->数组长度,i->当前需要维护的下标
var heapify = function (arr, n, i) {
        var largest = i;
        var lson = 2 * i + 1; //左节点
        var rson = 2 * i + 2; //右节点
        var temp
        if (lson < n && arr[largest] < arr[lson]) {
            largest = lson //如果左节点比当前节点大，largest=lson
        }
        if (rson < n && arr[rson] > arr[largest]) {
            largest = rson //如果右节点比当前节点大，largest=rson
        }
        //如果此时largest不等于i，说明左或者右节点大于当前下标，需要交换两个元素
        if (largest != i) {
            temp = arr[largest]
            arr[largest] = arr[i]
            arr[i] = temp
            heapify(arr, n, largest) //交换元素后需要继续维护当前元素的堆性质，递归调用heapify
        }
    }
    //堆排序入口 arr->数组,n->数组长度
    var heap_sort = function (arr, n) {
        //建堆
        for (var i = Math.floor(n/2)-1; i >= 0; i--) {
            heapify(arr, n, i) //通过heapify构建大顶堆
        }
        //排序
        for (var j = n - 1; j > 0; j--) {
        //此时第一个元素是最大的，进行排序操作，即第一个元素和最后一个元素进行交换
            var temp;
            temp = arr[0]
            arr[0] = arr[j]
            arr[j] = temp
            heapify(arr,j,0)
            //然后维护第一个元素的堆性质，断开最后一个元素和父节点的连接，所以heapify中
            //数组长度为j，需要维护堆性质的是arr[0]
        }
        return arr
    }