练习:最小的k个数

最小的k个数

1、题目要求:

输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。

2、我的代码:

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        if(input.size() < k) return vector<int>();
        vector<int> res;
        for(int i = 0;i < k;i++) {
            vector<int>::iterator iter = min_element(input.begin()+i, input.end());
            res.emplace_back(*iter);
            iter_swap(input.begin()+i, iter);
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/houzijushi/article/details/81140057