347. 前 K 个高频元素

最新推荐文章于 2024-09-17 11:50:15 发布

陈皮tang

最新推荐文章于 2024-09-17 11:50:15 发布

阅读量50

点赞数

分类专栏：算法刷题文章标签： java leetcode

本文链接：https://blog.csdn.net/tangxi666/article/details/120308534

版权

算法刷题专栏收录该内容

22 篇文章 0 订阅

订阅专栏

本文介绍如何使用HashMap存储输入数组中每个元素及其频率，然后借助小顶堆数据结构找到出现频率最高的前k个元素。通过代码实例演示了如何利用Java实现Solution类的topKFrequent方法。

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

class Solution {
    public int[] topKFrequent(int[] nums, int k) {

        HashMap<Integer,Integer> map = new HashMap<>();//数--频率
        for(int i=0;i<nums.length;i++){
            Integer count = map.get(nums[i]);
            if(count==null){
                map.put(nums[i],1);
            }else{
                map.put(nums[i],1+count);
            }
        }
        PriorityQueue<Map.Entry<Integer,Integer>> priority = new PriorityQueue<>((x,y)->(x.getKey()-y.getValue()));///用一个小顶堆，维护堆大小为k
        for(Map.Entry<Integer,Integer> entry : map.entrySet()){
            int value = entry.getValue();
            int key = entry.getKey();
            if(priority.size()<k){//堆的大小小于阈值
                priority.offer(entry);
            }else if(priority.peek().getValue()<value){//对的大小满足，且频率比对顶entry的频率value 要大
                priority.poll();
                priority.offer(entry);
            }
        }

        int[] res = new int[k];
        int i=0;
        for(Map.Entry<Integer,Integer> entry : priority){
            res[i++] = entry.getKey();
        }
        return res;
    }
}