剑指offer 面试题40. 最小的k个数 [简单]——快速排序

我的解题：

1.使用sort()对vector进行排序，将前k的元素插入vector<int> res中

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

2.快速排序

在快速排序中加入pivot左侧的元素数目与K的比较

需要注意的是，但需要k>这个数目，需要给数组的右侧进行排序是，k的值需要变化（卡在这里好久）

quicksort(nums,i+1,r,k-i+l-1)

class Solution {
public:
    void quicksort(vector<int>&nums,int l,int r,int k){
        if(l>=r) return;
        int i=l,j=r;
        int pivot=nums[l];
        while(i<j){
            while(nums[j]>=pivot&&i<j)  j--;
            if(i<j) nums[i++]=nums[j];
            while(nums[i]<=pivot&&i<j)  i++;
            if(i<j) nums[j--]=nums[i];
        }
        nums[i]=pivot;
        if(i-l+1==k)    return;
        if(k>i-l+1) quicksort(nums,i+1,r,k-i+l-1);
        else if(k<i-l+1)    quicksort(nums,l,i-1,k);
    }

    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> res;
        if(arr.empty()) return res;
        quicksort(arr,0,arr.size()-1,k);
        for(int i=0;i<k;i++)
            res.push_back(arr[i]);
        return res;
    }
};