Sword Finger Offer Interview Question 40.最小k数[シンプル]-クイックソート

私の解決策：

1. sort（）を使用してベクトルをソートし、最初のk要素をベクトル<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.クイックソート

ピボットの左側の要素数をクイックソートの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;
    }
};