最小的K个数（C++牛客网）

最新推荐文章于 2022-04-17 17:37:43 发布

coolsunxu

最新推荐文章于 2022-04-17 17:37:43 发布

阅读量141

点赞数

分类专栏： C++ 牛客网文章标签： c++ 数据结构算法 set 红黑树

本文链接：https://blog.csdn.net/coolsunxu/article/details/105772808

版权

C++ 同时被 2 个专栏收录

735 篇文章 18 订阅

订阅专栏

牛客网

209 篇文章 2 订阅

订阅专栏

解题思路：

（1）因为求解的是前k个最大的数，那么元素不能重合，使用set

（2）在插入的时候，顺便排序，最后取前k个

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        set<int> s; vector<int> v;
        if(k>input.size()) return v;
        for(auto w:input) s.insert(w);
        for(auto it=s.begin();k>0;it++,k--) v.push_back(*it);
        return v;
    }
};