leetcode 347. 前 K 个高频元素 中等 排序

题目：

分析：题目要返回频率前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;
            }
        }
    }
}