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

解：
①堆排序，priority_queue优先队列结构（C++默认大根堆）（也可以自己写实现）

class Solution {
public:
        //维护大小为k的优先队列（大顶堆）
        //C++默认是大顶堆
        priority_queue<int> a; 
        vector<int> res;

        for(int i = 0;i < arr.size();++i)
        {
            if(i < k) a.push(arr[i]);
            else if(arr[i] < a.top()) 
            {
                a.push(arr[i]);
                a.pop();
            }
            //a.push(arr[i]);
            //if(a.size() > k) a.pop();
        }

        while(!a.empty())
        {
            res.push_back(a.top());
            a.pop();
        }
        return res;

        /*sort(arr.begin(),arr.end());
        for(int i = 0;i < k;++i)
        {
            res.push_back(arr[i]);
        }
        return res;*/
};

②快排->返回数组无排序要求->快速选择

//找到第k大的数（索引k-1）
//随机一个数，调整顺序至左边的比它小，右边的比它大
//如果该数坐标为k-1，则它就为我们要找的数
//如果该数坐标小于k-1，在该数右边的数组中继续找；如果该数坐标大于k-1，在该数左边的数组中继续找。
//未调整队列：[p,r]

//排序，取前k个
//先存k个数进res,并维护一个
class Solution {
public:
    int k;
    vector<int> getLeastNumbers(vector<int>& arr, int k){

        if(arr.empty() || arr.size() < k || k < 1) return{};
        this->k = k;

        int resNum = arrAdjust(arr,0,arr.size() - 1); //得到调整数组后索引为k的位置
        
        vector<int> res(arr.begin(),arr.begin() + resNum);       
        return res;

    }
    int arrAdjust(vector<int>& arr,int p,int r)
    {
        if(p > r) return p; //防止arr.size() = k时，数组越界
        int val = arr[r];
        int low = p;
        int quick = p;

        while(quick < r)
        {
            if(arr[quick] < val) //记得访问数组。不要直接quick。
            {
                swap(arr,low,quick);
                low++;
            }
            quick++;
        }
        swap(arr,low,r);

        if(low == k) return low;
        else if(low < k) return arrAdjust(arr,low+1,r); //return 函数，不要直接调用函数
        else return arrAdjust(arr,0,low-1);
    }

    void swap(vector<int>& arr,int v1,int v2)
    {
        int tmp = arr[v1];
        arr[v1] = arr[v2];
        arr[v2] = tmp;
    }
};