js实现堆排序

堆，大顶堆

堆是一种特殊的完全二叉树，使用数组存储二叉树时，若某个非叶子节点存储在下标为i的位置，其左右孩子节点分别存储在下标为2i+1和2i+2的位置。

堆可以分为大顶堆和小顶堆，对大顶堆来说，任意非叶子节点不小于其左右孩子节点，对于小顶堆来说，任意非叶子节点不大于其左右孩子节点。若使用数组存储大顶堆，则满足：arr[i] >= arr[2i+1] && arr[i] >=arr[2i+2]（i为非叶子节点的在数组中的下标）

使用上述性质即可实现一个大顶堆：

实际上最核心的就是让第i项和他的子节点2i+1，2i+2项进行比较，交换位置等。其中handle函数的写法较为巧妙，通过判断上一级是否有交换来判断下一级是否与要交换。

        var heap = function(arr) {
            let k=arr.length
            let res=[]
            for (let i = 0; i <k; i++) {
                res.push(arr[i])
            }

            if (k!=1) {
                for(let i = Math.floor(k/2)-1; i>=0 ; i--) {
                    handle(res, k, i)
                }
            }

            return res
            
            function handle(arr,k,i) {
                while (true) {
                    let maxIndex=i
                    if (2*i+1<=k&&arr[2*i+1]>arr[i]) {
                        maxIndex=2*i+1
                    }
                    if (2*i+2<=k&&arr[2*i+2]>arr[maxIndex]) {
                        maxIndex=2*i+2
                    }
                    if (maxIndex!=i) {
                        swap(arr,maxIndex,i)
                        i = maxIndex
                    }else{
                        break
                    }
                }
                
            }
            function swap(arr, i , j){
                let temp = arr[i]
                arr[i] = arr[j]
                arr[j] = temp
            }
        };

 

应用于解题之中，如力扣剑指offer第40题，返回最小的k个数即可用大顶堆实现：

        var getLeastNumbers = function(arr, k) {
            
            let res=[]
            for (let i = 0; i <k; i++) {
                res.push(arr[i])
            }

            if (k!=1) {
                for(let i = Math.floor(k/2)-1; i>=0 ; i--) {
                    handle(res, k, i)
                }
            }

            for(let i = k; i < arr.length; i++) {
                if(res[0] > arr[i]) {
                    res[0] = arr[i]
                    handle(res, k, 0)
                }
            }

            return res
          
            function handle(arr,k,i) {
                while (true) {
                    let maxIndex=i
                    if (2*i+1<=k&&arr[2*i+1]>arr[i]) {
                        maxIndex=2*i+1
                    }
                    if (2*i+2<=k&&arr[2*i+2]>arr[maxIndex]) {
                        maxIndex=2*i+2
                    }
                    if (maxIndex!=i) {
                        swap(arr,maxIndex,i)
                        i = maxIndex
                    }else{
                        break
                    }
                }
                
            }
            function swap(arr, i , j){
                let temp = arr[i]
                arr[i] = arr[j]
                arr[j] = temp
            }
        };