题目:
分析:题目要返回频率前k高的元素,这句话就隐含了需要排序,排序后才能返回频率前k高的元素,要排序的话就要知道每个元素出现的个数,然后是排序算法的选择,像这种有关联值,值:频率 或 值:其它内容的要进行排序可以不选择一般的排序算法,因为一般排序算法没法关联另一个值,用Java的话可以使用优先队列,并且根据题目要求看是否需要自定义comparator(因为在排序器里可以在排序规则中关联其它值/属性)
优先队列排好序后直接循环k次取出答案即可
代码:
class Solution {
Map<Integer,Integer> count = new HashMap<>();
public List<Integer> topKFrequent(int[] nums, int k) {
for(int i = 0; i < nums.length; i++){
count.put(nums[i], count.getOrDefault(nums[i], 0) + 1);
}
Queue<Integer> countPQ = new PriorityQueue<>(new countPQComparator());
for(int key : count.keySet()){
countPQ.offer(key);
}
List<Integer> result = new ArrayList<>();
for(int i = 0; i < k; i++){
result.add(countPQ.poll());
}
return result;
}
class countPQComparator implements Comparator<Integer>{
//倒序排列
@Override
public int compare(Integer n1, Integer n2){
if(count.get(n1) < count.get(n2)){
return 1;
}else if(count.get(n1) > count.get(n2)){
return -1;
}else{
return 0;
}
}
}
}