Top K Frequent Elements

最新推荐文章于 2024-08-18 16:33:37 发布

沙虫zZ

最新推荐文章于 2024-08-18 16:33:37 发布

阅读量265

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/qq_17480053/article/details/51494103

leetcode 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种高效查找数组中最常出现的k个元素的方法。利用哈希表统计频率，并通过排序找出高频元素，确保时间复杂度优于O(n log n)。

Top K Frequent Elements

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表的使用，使用STL来解决比较方便。可以用map来记录词频，由于map不能sort，所以还需将map转换成pair然后排序，代码如下：

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> res;
        vector<pair<int,int>> pp;
        for(auto t:nums)res[t]++;
        for(auto r:res)pp.push_back(r);
        sort(pp.begin(),pp.end(),[=](pair<int,int>&a,pair<int,int>&b){return a.second>b.second;});
        vector<int> ans;
        for(int i = 0;i<k;i++)ans.push_back(pp[i].first);
        return ans;
    }
};