算法:求数组中最小的k个数,代码详解

剑指 Offer 40. 最小的k个数
输入整数数组 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]

限制:
0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000

基于快速排序的方法
总体思路,先对整个数组进行分区,获得划分分区的那个值的索引index,比较index和所求个数k的大小;
index > k:说明k个最小值都在分区的左部,继续对左部进行partition
index < k:说明k个最小值包含分区的左部所有值,并加上分区右部的部分值,所以继续对右部进行partition
index = k:说明index左部的值刚好k个,即所求的k个最小值


partition思路:(这个耗时较长)
先选择基准值,一般选最后一位,使用两个指针i,j,然后后从0开始遍历;
-[0,i)为小于基准值的数
-[i,j)为为遍历
-[j,end)为大于基准值的数
当前值<=基准值,i指针右移;
当前值>基准值,当前值和j指针所指数对调,j指针左移;
当指针i>j时,表示所有数遍历完成,结束遍历,然后把基准值和i所指的数对调

/**
 * @param {number[]} arr
 * @param {number} k
 * @return {number[]}
 */
var getLeastNumbers = function(arr, k) {
   
    //考虑特殊情况,当所求的数个数大于等于数组的长度,则之间返回数组即可
    if(arr.length <= k) return arr;
    //获得基准值的索引
    let index = partition(arr,0,arr.length-1);
    while
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值