347. 前 K 个高频元素

最新推荐文章于 2023-01-03 18:23:53 发布

小雨林夕

最新推荐文章于 2023-01-03 18:23:53 发布

阅读量94

点赞数

分类专栏：力扣笔记

本文链接：https://blog.csdn.net/qq_42283960/article/details/102716185

版权

力扣笔记专栏收录该内容

5 篇文章 0 订阅

订阅专栏

class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        HashMap<Integer,Integer> map = new HashMap<>();
        for(int num:nums){
            Integer count = map.get(num);
            if(count==null) count=0;
            map.put(num,count+1);
        }
        List<Integer>[] bucket = new ArrayList[nums.length+1];
        for(int key:map.keySet()){
            int count=map.get(key);
            if(bucket[count]==null){
                bucket[count]=new ArrayList<>();
            }
            bucket[count].add(key);
        }
        ArrayList<Integer> res = new ArrayList<Integer>();
        for(int i=bucket.length-1;i>=0 && res.size()<k;i--){
            if(bucket[i]!=null){
                res.addAll(bucket[i]);
            }
        }
        return res;
    }
}