【剑指Offer】面试题40.最小的k个数

面试题40.最小的k个数

题目

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

思路

 根据快速排序的启发，如果基于数组第k个数字进行调整，则使得比第k个数字小的所有数字都位于数组的左边，比第K个数字大的所有数字都位于数组的右边。
这样调整之后，位于数组中左边的k个数字就是最小的k个数字（这k个数字不一定是排序的）。

代码

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        if(k >= arr.size())  return arr;
        return quickSort(arr,k,0,arr.size() - 1);
    }

private:
    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);// 如果 k < i, 说明k + 1小的数字 存在于左子数组，则递归左子数组
        if(i < k) return quickSort(arr,k,i + 1,r);// 如果 k > i, 说明k + 1小的数字 存在于右子数组，则递归右子数组

        // 如果k = i, 代表arr[k]即为第k+1小的数字，则直接返回数组前k个数字即可
        vector<int>  res;
        res.assign(arr.begin(),arr.begin() + k);// 拷贝元素
        return res;
    }
};