剑指 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

    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        if(k==arr.size())
            return arr;
        return quicksort(arr,k,0,arr.size()-1);
    }
    vector<int> quicksort(vector<int>&arr,int k,int l,int r){
        int i=l,j=r;
        while(i<j){
            while(i<j&&arr[j]>=arr[l])
                j--;
            while(i<j&&arr[i]<=arr[l])
                i++;
            swap(arr[i],arr[j]);
        }
        swap(arr[i],arr[l]);
        if(i>k)
            return quicksort(arr,k,l,i-1);
        if(i<k)
            return quicksort(arr,k,i+1,r);
        vector<int>res;
        res.assign(arr.begin(),arr.begin()+k);
        return res;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值