给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
示例 1:
**输入:**nums = 1,1,1,2,2,3, k = 2
输出:1,2
示例 2:
**输入:**nums = 1, k = 1
输出:1
示例 3:
**输入:**nums = 1,2,1,2,1,2,3,1,3,2, k = 2
输出:1,2
提示:
1 <= nums.length <= 105-104 <= nums[i] <= 104k的取值范围是[1, 数组中不相同的元素的个数]- 题目数据保证答案唯一,换句话说,数组中前
k个高频元素的集合是唯一的
进阶: 你所设计算法的时间复杂度 必须 优于 O(n log n) ,其中 n是数组大小。
java
class Solution {
public int[] topKFrequent(int[] nums, int k) {
//转换,将值和次数对应起来
Map<Integer,Integer> map=new HashMap<>();
for(int num:nums){
map.put(num,map.getOrDefault(num,0)+1);
}
//前k个加入优先队列,过则淘汰,自定义规则,实现Comparator接口
PriorityQueue<int[]> queue=new PriorityQueue<>(new Comparator<int[]>(){
public int compare(int[] m,int[] n){
return m[1]-n[1];
}
});
//便利map加入queue
for(Map.Entry<Integer,Integer> entry:map.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});
}
}
//取出queue
int[] res=new int[k];
for(int i=0;i<k;i++){
res[i]=queue.poll()[0];
}
return res;
}
}