347. 前 K 个高频元素
时间:2020年9月7日
知识点:hashmap、最小堆
题目链接:https://leetcode-cn.com/problems/top-k-frequent-elements/
题目
给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
示例1
输入:
nums = [1,1,1,2,2,3], k = 2
输出:
[1,2]
示例2
输入:
nums = [1], k = 1
输出:
[1]
说明:
- 你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
- 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
- 题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的。
- 你可以按任意顺序返回答案。
解法
- 首先想到用unordered_map 建立数字和个数的映射
- 对频率排序
- 对频率排序有两种方法,
- 把map转成vector,包含数字和个数,全部排列,取前k个,时间复杂O(N2)
- 把map转成大小为堆,堆的大小为k,时间复杂度为O(Nlogk)
- 取出频率对应的数字
代码
#include <stdio.h>
#include <iostream>
#include <vector>
#include <unordered_map>
#include <queue>
using namespace std;
/*
//hashmap+快排
bool cmp(pair<int,int> x, pair<int,int> y){
return x.second>y.second;
};
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
vector<int> ans;
unordered_map<int,int> m;
int n = nums.size();
vector<pair<int,int>> v(n);
for(int i=0;i<nums.size();i++){
m[nums[i]] ++;
}
for(unordered_map<int, int>::iterator i=m.begin();i!=m.end();i++){
v.push_back(*i);
}
sort(v.begin(),v.end(),cmp);
for(int i=0;i<k;i++)
ans.push_back(v[i].first);
return ans;
}
};
*/
//hashmap+堆
class Solution {
public:
struct cmp{
bool operator()(pair<int, int> x,pair<int, int> y){
return x.second > y.second;
}
};
vector<int> topKFrequent(vector<int>& nums, int k) {
vector<int> ans(k,0);
unordered_map<int,int> m;
priority_queue<pair<int, int>,vector<pair<int, int>>,cmp> q;
int n = nums.size();
for(int i=0;i<nums.size();i++){
m[nums[i]] ++;
}
for(auto i=m.begin();i!=m.end();i++){
q.push(*i);
if(q.size() > k)
q.pop();
}
for(int i=k-1;i>=0;i--){
ans[i] = q.top().first;
q.pop();
}
return ans;
}
};
int main()
{
vector<int> nums{1,1,1,2,2,3};
Solution s;
int k = 2;
vector<int> ans = s.topKFrequent(nums, k);
for(int i=0;i<k;i++)
cout<<ans[i]<<endl;
return 0;
}
今天也是爱zz的一天哦!