看到这道题一开始的思路是:使用map:键存放nums,值存放nums出现的次数。然后通过对值排序的方法找出前k个元素。这样也能做,代码如下:
class Solution {
public int[] topKFrequent(int[] nums, int k) {
Map<Integer,Integer> map=new HashMap<>();
int [] res=new int[k];
for(int i=0;i<nums.length;i++){
map.put(nums[i],map.getOrDefault(nums[i],0)+1);
}
List<Map.Entry<Integer, Integer>> entries = new ArrayList<>(map.entrySet());
Collections.sort(entries, (e1, e2) -> e2.getValue().compareTo(e1.getValue()));
for (int i = 0; i < k && i < entries.size(); i++) {
res[i]=entries.get(i).getKey();
}
return res;
}
}
官方题解给出的是用优先级队列做,而且用的是小顶堆。大顶堆的话,每次移除的都是最大的元素(×)
代码如下:
class Solution {
public int[] topKFrequent(int[] nums, int k) {
int [] res=new int[k];
Map<Integer,Integer> map=new HashMap<>();
for(int i=0;i<nums.length;i++){
map.put(nums[i],map.getOrDefault(nums[i],0)+1);
}
PriorityQueue<Integer> pq=new PriorityQueue<>(new Comparator<Integer>(){
public int compare(Integer a,Integer b){
return map.get(a)-map.get(b);
}
});
for(Integer key:map.keySet()){
if(pq.size()<k) pq.add(key);
else if(map.get(key)>map.get(pq.peek())){
pq.remove();
pq.add(key);
}
}
for(int i=0;i<k;i++){
res[i]=pq.remove();
}
return res;
}
}