347. Top K Frequent Elements

最新推荐文章于 2024-06-11 11:37:12 发布

原创最新推荐文章于 2024-06-11 11:37:12 发布 · 226 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#优先队列

leetCode 专栏收录该内容

171 篇文章

订阅专栏

本文介绍了一种高效算法，用于从整数数组中找出出现频率最高的k个元素。该算法利用哈希表记录每个元素出现的次数，并通过优先队列（小顶堆）保持当前k个最频繁元素。此方法优于O(nlogn)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


/*
Given a non-empty array of integers, return the k most frequent elements.

For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].

Note: 
You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm's time complexity must be better than O(n log n), where n is the array's size.
使用一个hash表对重复的数字计数，使用优先队列对hash中的计数值进行排序
*/

#include <iostream>
#include <vector>
#include <unordered_map>
#include <queue>
#include <utility>
using namespace std;

class Solution {
private:
    struct bigger{
        bool operator()(pair<int,int> &a,pair<int,int>&b)
        {
            return a.first > b.first;
        }
    };
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> mp;
        for(int i=0;i<nums.size();i++)
            mp[nums[i]]++;
        vector<int> res;
        priority_queue<pair<int,int>,vector<pair<int,int>>,bigger> p;//小顶堆
        for(unordered_map<int,int>::iterator it=mp.begin();it!=mp.end();it++)
        {
            if(p.size()<k)
                p.push(make_pair(it->second,it->first));
            else if(it->second >= p.top().first)
            {
                p.pop();
                p.push(make_pair(it->second,it->first));
            }
        }
        while(!p.empty())
        {
            res.push_back(p.top().second);
            p.pop();
        }
        return res;
    }
};
int main()
{
    Solution mys;
    vector<int> nums={2,3,4,3,2,3};
    int k=2;
    vector<int>res=mys.topKFrequent(nums,k);
    for(int i=0;i<res.size();i++)
        cout<<res[i]<<endl;

    return 0;
}