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允許