347. 前 K 个高频元素

最新推荐文章于 2024-10-01 22:51:02 发布

三金samkam

最新推荐文章于 2024-10-01 22:51:02 发布

阅读量115

点赞数

分类专栏： LeetCode 文章标签：散列表数据结构 LeetCode leetcode java

本文链接：https://blog.csdn.net/qq_39715243/article/details/124566627

版权

LeetCode 专栏收录该内容

53 篇文章 1 订阅

订阅专栏

该博客讨论了一种使用优先级队列（PriorityQueue）在Java中找到整数数组中出现频率最高的k个元素的方法。代码实现中，首先通过哈希表统计每个元素的出现次数，然后创建一个基于频率比较的优先级队列。当队列满时，新元素如果频率更高则替换队列中的低频元素。最终，从队列中取出前k个高频元素作为结果。

摘要由CSDN通过智能技术生成

链接：力扣

题目：

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

示例：

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

代码：

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> occurences = new HashMap<Integer, Integer>();
        //哈希表存储num的各个数的个数
        for(int num:nums){
            occurences.put(num, occurences.getOrDefault(num, 0) + 1);
        }

        //基于优先级的无界优先级队列
        // int[] 的第一个元素代表数组的值，第二个元素代表了该值出现的次数
        PriorityQueue<int[]> queue = new PriorityQueue<int[]>(new Comparator<int[]>() {
            public int compare(int[] m, int[] n) {
                return m[1] - n[1];
            }
        });

        for(Map.Entry<Integer, Integer> entry : occurences.entrySet()){
            int num = entry.getKey(), count = entry.getValue();
            if (queue.size() == k) {
                //发现新的数量多的元素，在队列里弹出数量少的
                if (queue.peek()[1] < count) {
                    queue.poll();
                    queue.offer(new int[]{num, count});
                }
            } else {
                queue.offer(new int[]{num, count});
            }

        }
        
        int[] ret = new int[k];
        for (int i = 0; i < k; ++i) {
            ret[i] = queue.poll()[0];
        }
        return ret;


    }
}