返回最小的k个数

Posted 2020-09-30 王小东大将军

对于一个无序数组，数组中元素为互不相同的整数，请返回其中最小的k个数，顺序与原数组中元素顺序一致。

给定一个整数数组A及它的大小n，同时给定k，请返回其中最小的k个数。

测试样例：

[1,2,4,3],4,2

返回：[1,2]

1、排序

2、找到k大作为基准

3、遍历比较，小于k的输出

vector<int> findKthNumbers(vector<int> A, int n, int k) {
        vector<int> B = A;
        vector<int> ans;
        sort(B.begin(),B.end());
        for(int i=0;i<A.size();i++)
            if(A[i] <= B[k-1])
              ans.push_back(A[i]);
        return ans;
    }