最小的K个数

题目链接

Solution

求最小的k个数。
首先肯定是将数组从小到大排序，不然不好选择。
然后从头开始取k个数即可。

Code

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        sort(input.begin(), input.end());
        vector<int> ans;
        if (k > input.size()) return ans;
        for (int i = 0; i < k; ++i) ans.push_back(input[i]);
        return ans;
    }
};