class Solution {
public int[] topKFrequent(int[] nums, int k) {
HashMap<Integer,Integer> map = new HashMap<>();
PriorityQueue<Map.Entry<Integer,Integer>> queue = new PriorityQueue<>(new Comparator<Map.Entry<Integer,Integer>>(){
@Override
public int compare(Map.Entry<Integer,Integer> o1,Map.Entry<Integer,Integer> o2){
return o2.getValue() - o1.getValue();
}
});
int[] res = new int[k];
for(int i=0;i<nums.length;i++){
map.put(nums[i],map.getOrDefault(nums[i],0)+1);
}
Set<Map.Entry<Integer,Integer>> set=map.entrySet();
for(Map.Entry<Integer,Integer> s:set){
queue.add(s);
}
int j=0;
for(int i=0;i<k;i++){
res[j++]=queue.poll().getKey();
}
return res;
}
}
需要注意的地方:
1.注意这个Map.Entry<>,没有构造方法,是个接口
2.对于这道题而言,频率正是意味着一种优先级,对于优先级队列而言,正合适,但是怎么更好使用呢?可以不断更改每个元素的优先级呢?元素就是Key,优先级大小就是Value
3.因为Map.Entty不能直接构造,只能通过Map来put,所以先形成HashMap,在一个一个加入队列
4.HashMap的entrySet可以用来得到Map.Entty的集合
5.如果只获得Map的value集合,排序后,无法得到对应的Key,Key不允许重复,Value允许