LeetCode:347. 前 K 个高频元素

最新推荐文章于 2022-07-05 16:11:56 发布

鸡蛋饺子粥

最新推荐文章于 2022-07-05 16:11:56 发布

阅读量92

点赞数

分类专栏： # 力扣&牛客文章标签： leetcode 算法 java

本文链接：https://blog.youkuaiyun.com/qq_40859392/article/details/124773587

版权

力扣&牛客专栏收录该内容

106 篇文章

订阅专栏

该博客介绍了如何利用优先级队列（PriorityQueue）在Java中实现对整数数组的元素进行频率排序，找出出现频率最高的k个元素。代码示例中，首先使用HashMap统计每个元素的出现次数，然后将元素和其频率作为数组插入优先级队列，队列按照频率降序排列。最后，从队列中取出前k个元素作为结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述（原题链接）

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

代码：

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        //优先级队列
        PriorityQueue<int[]> q = new PriorityQueue<int[]>(new Comparator<int[]>(){
            public int compare(int[] o1,int[] o2){
                return o2[1]-o1[1];
            }
        });
        HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
        for(int i=0;i<nums.length;i++){
            map.put(nums[i],map.getOrDefault(nums[i],0)+1);
        }
        for(int i:map.keySet()){
            q.offer(new int[]{i,map.get(i)});
        }
        int[] res = new int[k];
        for(int i=0;i<k;i++)
        res[i]=q.poll()[0];
        return res;
    }
}