给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2 输出: [1,2]
示例 2:
输入: nums = [1], k = 1 输出: [1]
解析:
这道题目主要涉及到如下三块内容:
- 要统计元素出现频率
- 对频率排序
- 找出前K个高频元素
首先统计元素出现的频率,这
一类的问题可以使用map来进行统计。
然后是对频率进行排序,这里我们可以使用一种 容器适配器就是优先级队列。
//出现频率最多的前k个
public int[] topKFrequent(int[] nums,int k){
HashMap<Integer, Integer> hashMap = new HashMap<>();
//1. 先使用hashmap
for (int i=0;i<nums.length;i++){
//key:nums[i] value:出现频率:
hashMap.put(nums[i],hashMap.getOrDefault(nums[i],0)+1);
}
// 2.使用优先级队列;存入的是一个数组 ;需要自定义比较器:按元素出现频率升序的方式排列
PriorityQueue<int[]> queue = new PriorityQueue<>((num1, num2) -> num1[1] - num2[1]);
//遍历hashmap ---->一个键值对实体:
for (Map.Entry<Integer, Integer> entry: hashMap.entrySet()) {
if (queue.size()<k){
//如果队列中没有放入值
queue.add(new int[]{entry.getKey(),entry.getValue()});
}else{
//如果队列中已经放入了k个元素;
//如果放入的值大于队头中元素出现的次数 ,就要进行比较将较小的进行出队
if (entry.getValue()>queue.peek()[1]){
queue.poll();//出栈;
//加入新的值:
queue.add(new int[]{entry.getKey(), entry.getValue()});
}
}
}
//3.倒序插入前k个数
int[] res = new int[k];
//出现频率越多的数,越先优先出现数组的前面,所以要倒序放置元素;使出现次数少的元素排在后面
for (int i=res.length-1;i>=0;i--){
res[i]=queue.poll()[0];
}
return res;
}