LeetCode最小的k个数

最新推荐文章于 2024-01-23 09:34:00 发布

yqtaowhu

最新推荐文章于 2024-01-23 09:34:00 发布

阅读量2.5k

点赞数

分类专栏： Leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/taoyanqi8932/article/details/51768825

版权

Leetcode 专栏收录该内容

113 篇文章 0 订阅

订阅专栏

/************************************************************************
*    Author : yqtao
*    Email  :yqtao@whu.edu.cn
*    Date   :2016.6.27
*************************************************************************/
#include <set>
#include<iostream>
#include<vector>
#include <functional>
using namespace std;
multiset<int,greater<int>> leastKNums(vector<int> &nums, int k)
{

    multiset<int, greater<int>> res;
    if (nums.size() < k)      
    {
        multiset<int, greater<int>>res(nums.begin(), nums.end()); 
        return res;
    }
    if (k < 1)
        throw "error";
    for (auto it = nums.begin(); it != nums.end();++it)  //遍历需要处理的数组
    {
        if (res.size() < k)
            res.insert(*it);
        else
        {
            auto iter = res.begin();//即k个数中的最大值
            if (*it < *iter)
            {
                res.erase(iter);
                res.insert(*it);
            }
        }
    }
    return res;
}
int main()
{
    vector<int>nums = { 1,2,3,4,5,6,7,8,9,3,2,4,6,7,6,56,4354,324, };
    multiset<int, greater<int>> res = leastKNums(nums, 10);
    for (auto c : res)
        cout << c << " " << endl;
    return 0;
}