leetcode-最小的k个数

1、题目

给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。
数据范围：0≤k,n≤10000，数组中每个数的大小0≤val≤1000
要求：空间复杂度 O(n) ，时间复杂度 O(nlogn)

2、思路

用选择排序的方法，排序到第k小的时候就结束排序。

3、代码

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        choicesort(input, k);
        vector<int> res;
        for(int i=0; i<k; i++){
            res.push_back(input[i]);
        }
        return res;
    };
    
    void choicesort(vector<int> &arr, int k){
        for(int i = 0; i < k; i++){
            int minindex = i;
            for(int j = i+1; j < arr.size(); j++){
                if(arr[j]<arr[minindex]){
                    minindex = j;
                }
            }
            this->swap(arr, i, minindex);
        };
        
    };
    void swap(vector<int> &arr, int i, int j){
        int tmp = arr[i];
        arr[i] = arr[j];
        arr[j] = tmp;
    }
};