leetcode算法之堆(简单)

*输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
示例 1:

输入:arr = [3,2,1], k = 2
输出:[1,2] 或者 [2,1]

示例 2:

输入:arr = [0,1,2,1], k = 1
输出:[0]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。*

/**
 * @param {number[]} arr
 * @param {number} k
 * @return {number[]}
 */
/* var getLeastNumbers = function(arr, k) {
    //先给arr从小到大排序
    let newarr=arr.sort(function sortnum(a,b){
        return a-b;
    });
    return newarr.slice(0,k);

};*/
var getLeastNumbers = function(arr, k) {
    let len = arr.length
    if (!len || !k) return []
    let start = 0
    let end = len - 1
    // 寻找一次标杆元素的位置
    let index = quikSort(arr, start, end)
    // 如果标杆元素的位置不等于 K
    while(index !== k - 1) {
        if (index > k-1) {
            // 如果上一次查找,标杆元素位置大于目标位置
            end = index-1
            index = quikSort(arr, start, end)
        } else {
            // 如果上一次查找,标杆元素位置小于目标位置
            start = index + 1
            index = quikSort(arr, start, end)
        }
    }
    return arr.slice(0, index+1)
};

function quikSort(arr, left, right) {
    let pivot = arr[left]
    while(left < right) {
        while(left < right && arr[right] >= pivot) right--;
        [arr[left],arr[right]] = [arr[right],arr[left]];
        while(left < right && arr[left] < pivot) left++
        [arr[left],arr[right]] = [arr[right],arr[left]];
    }
    arr[left] = pivot
    return left
}

分析:快速排序只需要把前k个排列好就可以
关于快速排序:分别由low high两个标还有一个temp存储标准,当low的值大于high和且low对应的值temp对应的值两者交换否则low–;另一种情况high++。
存储的标准左右两边分别是比他大或小的元素,具体大小要看是从大到小排列还是从小到大排列

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值