LeetCode: Top K Frequent Elements

最新推荐文章于 2024-10-20 08:13:14 发布

原创最新推荐文章于 2024-10-20 08:13:14 发布 · 248 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #Top K Frequent Eleme #map

LeetCode 专栏收录该内容

71 篇文章

订阅专栏

本文介绍了一种高效查找数组中出现频率最高的k个数的方法。利用哈希表统计每个元素出现次数，并通过多重映射实现按频次排序。该算法优于传统的排序统计方法，时间复杂度达到O(n)级别。

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 logn), wheren is the array's size.

题目解析：找出一个数组中出现频率最高的k个数

算法：这里可以先排序再统计，时间复杂度为O(nlog(n))

源代码：

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> cnt;
        unordered_map<int, int>::iterator it;
        int len = nums.size();
        
        for (int i = 0; i < len; ++i) {
            cnt[nums[i]]++;
        }
        
        multimap<int, int, greater<int> > mmp;
        for (it = cnt.begin(); it != cnt.end(); ++it) {
            mmp.insert(pair<int, int>(it->second, it->first));
        }
        
        vector<int> res(k, 0);
        multimap<int, int, greater<int> >::iterator mit;
        int i = 0;
        for (i = 0, mit = mmp.begin(); i < k; ++i, ++mit) {
            res[i] = mit->second;
        }
        
        return res;
        
    }
};