Top K Frequent Elements 前K个高频元素

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

示例 1:

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

示例 2:

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

说明:

  • 你可以假设给定的 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
  • 你的算法的时间复杂度必须优于 O(n log n) , 是数组的大小。

思路:先把数据按照<数字,数字对应次数>放到map里,然后把map的value放到最大堆中(相当于对value进行排序,由大到小排序),最后输出前k个元素。

参考代码:

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
	vector<int> res;
	priority_queue<pair<int,int> > heap;
	unordered_map<int, int> map;
	for (auto num : nums) map[num]++;
	for (auto it: map) heap.push({ it.second,it.first});
	for (int i = 0; i < k; i++) {
		res.push_back(heap.top().second); heap.pop();
	}
	return res;        
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值