347. 前 K 个高频元素(leetcode)

https://leetcode.cn/problems/top-k-frequent-elements/description/

可以考虑使用HashMap+排序,不过直接使用优先队列也挺不错,可以使用大顶堆或小顶堆

class Solution {
    
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        for(int n:nums){
            map.put(n,map.getOrDefault(n,0)+1);
        }
        // int[] 第一个元素是数值,第二个元素是频次, 重写的比较规则是大顶堆
        PriorityQueue<int[]> pq = new PriorityQueue<>((pair1,pair2)-> pair2[1]-pair1[1]);
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            pq.add(new int[]{entry.getKey(), entry.getValue()});
        }
        int[] res=new int[k];
        for(int i=0;i<k;i++){
            res[i]=pq.poll()[0];
        }
        return res;

    }
}