[剑指offer]最小的k个数

[剑指offer]最小的k个数

剑指offer-最小的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

解题思路

• 快速选择算法，快速排序的partition函数会返回一个pos，使得arr[0……pos-1]<=arr[pos]，arr[0……pos-1]>=arr[pos]，那么它就是数组第pos大的数。
• 直接用sort排序。。。

//快速选择
class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> res;
        int len=arr.size();
        if(len==0||k>len||k<=0)
            return res;
        int l=0;
        int h=len-1;         
        while(l<h){
            int pos=partition(arr,l,h);
            if(pos==k-1)
                break;
            else if(pos>k-1)
                h=pos-1;
            else
                l=pos+1;
        }
        for(int i=0;i<k;i++)
            res.push_back(arr[i]);
        return res;
    }
    
    int partition(vector<int>& arr,int start,int end){
        int flag=arr[start];
        int left=start+1;
        int right=end;
        while(left<=right){
            while(left<=end&&arr[left]<flag)
                left++;
            while(right>=0&&arr[right]>flag)
                right--;
            if(left<=right)
                swap(arr[left++],arr[right--]);
        }
        swap(arr[right],arr[start]);
        return right;
    }
};

//sort
class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> res;
        if(arr.size()==0)
            return res;
        sort(arr.begin(),arr.end());
        for(int i=0;i<k;i++)
            res.push_back(arr[i]);
        return res;
    }
};