[LeetCode]347. 前 K 个高频元素(堆)
2024-10-09 14:42:29
题目
给定一个非空的整数数组,返回其中出现频率前 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 是数组的大小。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解
- 先遍历一遍记录各元素出现次数。
- 使用小根堆维护前k个频率最高元素(因为超过k个需要当前元素和堆顶元素比较,决定去掉哪个,所以使用小根堆而不是大根堆),其中定义比较器使用根据HashMap得到的元素频率大小比较。
代码
class Solution {
public List<Integer> topKFrequent(int[] nums, int k) {
HashMap<Integer, Integer> map = new HashMap<>();
for (int num : nums) {
map.put(num, map.getOrDefault(num, 0) + 1);
}
PriorityQueue<Integer> minHeap = new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return map.get(o1) - map.get(o2);
}
});
for (int key : map.keySet()) {
minHeap.add(key);
if (minHeap.size() > k) {
minHeap.poll();
}
}
List<Integer> list = new LinkedList<>();
for (int num : minHeap) {
list.add(num);
}
return list;
}
}
最新文章
- 2013 duilib入门简明教程 -- 简单控件介绍 (12)
- MYSQL 中 update set from where 问题
- 分享2个网址二维码API接口
- OGNL_一点
- (转)CSS 为不同大小的浏览器视窗使用不同的样式表
- C# Winform 支持Hex与ASCII输入和切换的文本框
- C语言经典算法100例
- Eclipse无提示的解决办法 和 内容辅助技巧
- SQL Server 2012 - 内置函数
- linux系统盘使用率达到100%的问题查找和解决方法
- html细节积累-01
- VC下ffmpeg例程调试报错处理
- Vue--学习过程中遇到的坑
- 转 jQuery中的$.extend方法来扩展JSON对象
- STL之set和map
- 一脸懵逼学习Struts数据校验以及数据回显,模型驱动,防止表单重复提交的应用。
- FileReader &; Blob &; File
- php分享十八七:mysql基础
- python之路----面向对象进阶一
- python模块之pyMySql
热门文章
- Goland 生成可执行文件
- 如何解决Mybatis 日期查询时后面带.0
- MySQL 偶尔抽风,性能突然下降
- SSM框架整合练习——一个简单的文章管理系统
- JAVA使用urlrewrite实现伪静态化
- golang 创建 tun 设备
- selenium入门编程总结学习于龙腾)
- 四则运算生成命令行程序 (Python)
- c++: internal compiler error: Killed (program cc1plus)
- akka-grpc - 基于akka-http和akka-streams的scala gRPC开发工具