1. 程式人生 > >返回最小的k個數

返回最小的k個數

urn begin 其中 基準 light 比較 logs for push_back

對於一個無序數組,數組中元素為互不相同的整數,請返回其中最小的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;
    }

  

返回最小的k個數