ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

347前 K 个高频元素(哈希表、堆排序)

2021-01-03 14:59:13  阅读:194  来源: 互联网

标签:347 map res 元素 堆排序 iter 计数 vector 哈希


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

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有