347. 前K个高频元素

前K个高频元素

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。
示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

输入: nums = [1], k = 1
输出: [1]

思路+代码+注释：

public List<Integer> topKFrequent(int[] nums, int k) {
            /*
            思路：使用hashMap统计每个数字出现的次数，然后使用堆排序堆次数进行排序，求出Top k
             */
            List<Integer> res=new ArrayList<>();
        Map<Integer,Integer> map=new HashMap<>();
        for (int i:nums
             ) {
            if (map.containsKey(i))
            {
                map.put(i,map.get(i)+1);
            }else {
                map.put(i,1);
            }
        }
        PriorityQueue<Map.Entry> priorityQueue=new PriorityQueue<>(new Comparator<Map.Entry>() {
            @Override
            public int compare(Map.Entry o1, Map.Entry o2) {
                return (Integer)o2.getValue()-(Integer) o1.getValue();
            }
        });
        priorityQueue.addAll(map.entrySet());
        for (int i = 0; i < k; i++) {
            res.add((Integer) priorityQueue.poll().getKey());
        }
        return res;
    }