*最小的k个数(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]

1.sort,冒泡排序

var getLeastNumbers = function(arr, k) {
    arr.sort((a, b)=> a-b)
    return arr.slice(0, k)
};

2.优化(快排)

/**
 * @param {number[]} arr
 * @param {number} k
 * @return {number[]}
 */
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]
        while(left < right && arr[left] < pivot) left++
        arr[right] = arr[left]
    }
    arr[left] = pivot
    return left
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值