347前 K 个高频元素（哈希表、堆排序）

1、题目描述

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

提示：

• 你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
• 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
• 你可以按任意顺序返回答案。

2、示例

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

3、题解

解法一：

基本思想：小顶堆，构造前k个最大频率元素的小顶堆

解法二：

基本思想：哈希表（好的数据结构是成功的一半），先对元素对应计数映射得到map cnt，然后反向计数对应元素映射得到map Map,因为可能同一个计数对应多个元素所以value是vector<int>，因为map是红黑树，所以已经排序好了，只要从尾到头取k个元素即可。

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        //基本思想：小顶堆，构造前k个最大频率元素的小顶堆
        vector<int> res;
        map<int,int> cnt;  //元素计数，key是元素，value是计数
        priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> min_heap;
        for(int i=0;i<nums.size();i++)
            cnt[nums[i]]++;
        for(auto iter=cnt.begin();iter!=cnt.end();iter++)
        {
            min_heap.push(pair<int,int>(iter->second,iter->first));
            if(min_heap.size()>k)
                min_heap.pop();
        }
        while(!min_heap.empty())
        {
            res.push_back(min_heap.top().second);
            min_heap.pop();
        }
        return res;
    }
};
class Solution1 {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        //基本思想：哈希表，先对元素对应计数映射得到map cnt，然后反向计数对应元素映射得到map Map
        //因为可能同一个计数对应多个元素所以value是vector<int>，因为map是红黑树，所以已经排序好了，只要从尾到头取k个元素即可
        vector<int> res;
        map<int,int> cnt;  //元素对应计数映射
        map<int,vector<int>> Map;  //计数对应元素映射
        for(int i=0;i<nums.size();i++)
            cnt[nums[i]]++;
        for(auto iter=cnt.begin();iter!=cnt.end();iter++)
            Map[iter->second].push_back(iter->first);
        int n=0;
        for(auto iter=Map.rbegin();iter!=Map.rend()&&n<k;iter++)
        {
            res.insert(res.end(),(iter->second).begin(),(iter->second).end());
            n+=(iter->second).size();
        }
        return res;
    }
};
int main()
{
    Solution solute;
    vector<int> nums={5,2,5,3,5,3,1,1,3};
    int k=2;
    vector<int> res=solute.topKFrequent(nums,k);
    for_each(res.begin(),res.end(),[](auto v){cout<<v<<endl;});
    return 0;
}

标签：347,map,res,元素,堆排序,iter,计数,vector,哈希
来源： https://blog.csdn.net/Revendell/article/details/112134943