leetcode347 Top K Frequent Elements java

最新推荐文章于 2025-04-21 22:09:55 发布

店小不二

最新推荐文章于 2025-04-21 22:09:55 发布

阅读量344

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode java

本文链接：https://blog.youkuaiyun.com/u013190088/article/details/66995597

leetcode 专栏收录该内容

41 篇文章

订阅专栏

本文介绍了一种高效算法来找出数组中出现频率最高的k个元素。通过使用哈希映射记录每个元素出现的次数，并利用桶排序的思想，将相同出现次数的元素归类到同一数组下标对应的桶中，从而实现O(n)时间复杂度内解决问题。

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

欢迎转载，转载请注明原地址：
http://blog.youkuaiyun.com/u013190088/article/details/66995597

Description

Given a non-empty array of integers, return the k most frequent elements.

For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2]
Note:
You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm’s time complexity must be better than O(n log n), where n is the array’s size.

解法：

首先想到的就是用map存储数字以及其出现的次数。之后将次数作为数组下标，根据次数把数字分到若干个“桶”（数组中的每个位置都相当于一个“桶”）里边，最后从后向前遍历数组即可。

public List<Integer> topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        List<Integer>[] arr = new List[nums.length + 1];
        for(int n : nums) {
            map.put(n, map.getOrDefault(n,0) + 1);
        }
        for(int key : map.keySet()) {
            int frequency = map.get(key);
            if(arr[frequency] == null) arr[frequency] = new ArrayList<>();
            arr[frequency].add(key);
        }
        List<Integer> res = new ArrayList<>();
        for(int j=arr.length-1; res.size()<k; j--) {
            if(arr[j] != null) {
                res.addAll(arr[j]);
            }
        }

        return res;
    }