- 作者:老汪软件技巧
- 发表时间:2024-01-06 07:00
- 浏览量:
347.前K个高频元素
方法:堆
首先遍历整个数组,并使用哈希表记录每个数字出现的次数,并形成一个「出现次数数组」。找出原数组的前 k 个高频元素,就相当于找出「出现次数数组」的前 k 大的值
利用堆的思想:建立一个小顶堆,然后遍历出现次数数组:
class Solution {
public int[] topKFrequent(int[] nums, int k) {
//使用字典,统计每个元素出现的次数,元素为键,元素出现的次数为值
Map occurrences = new HashMap();
for(int num:nums){
occurrences.put(num,occurrences.getOrDefault(num,0) + 1);
}
//int[] 的第一个元素代表数组的值,第二个元素代表了该值出现的次数
PriorityQueue queue = new PriorityQueue(new Comparator() {
public int compare(int[] m,int[] n) {
return m[1] - n[1];
}
});
for(Map.Entry entry : occurrences.entrySet()){
int num = entry.getKey(),count = entry.getValue();
if(queue.size() == k){
if(queue.peek()[1] < count){
queue.poll();
queue.offer(new int[]{num,count});
}
}else{
queue.offer(new int[]{num,count});
}
}
//取出堆中的元素
int[] ret = new int[k];
for(int i = 0;i