347.前 K 个高频元素

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

题目要求

给你一个整数数组 nums和一个整数 k ，请你返回其中出现频率前 k高的元素。你可以按 任意顺序返回答案。

优先队列

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        int n = nums.length;
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        PriorityQueue<int[]> queue = new PriorityQueue<>(k, (o1, o2) -> o2[1] - o1[1]);
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            queue.add(new int[]{entry.getKey(), entry.getValue()});
        }
        int[] res = new int[k];
        for (int i = 0; i < k && !queue.isEmpty(); i++) {
            res[i] = queue.poll()[0];
        }
        return res;
    }
}

思路

• 使用优先队列实现大顶堆，自定义排序规则，让队列中的元素都按照出现次数从大到小进行排列，输出前k个元素即可；