出现频率最多的 k 个元素

题目

题源

代码

class Solution {
    /**
    先通过hashMap存储，键是num，值是频次；
    接下来，存储到桶中，桶的下标是频次，桶中存入多个频次相同的num；
    再次，遍历桶，从后往前遍历，如果该桶有数则加入ArrayList中；
    最后，将arrayList中的num转为int数组。
     */
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> hashMap = new HashMap<>();
        for (int num : nums) {
            hashMap.put(num, hashMap.getOrDefault(num, 0) + 1);
        }
        List<Integer>[] buckets = new ArrayList[nums.length + 1];
        for(int key : hashMap.keySet()) {
            int val = hashMap.get(key);
            if (buckets[val] == null) {
                buckets[val] = new ArrayList<>();
            }
            buckets[val].add(key);
        }
        List<Integer> list = new ArrayList<>();
        for (int i = buckets.length - 1; i >= 0 && list.size() < k; i--) {
            if (buckets[i] == null) continue;
            if (buckets[i].size() <= (k - list.size())) {
                list.addAll(buckets[i]);
            } else {
                list.addAll(buckets[i].subList(0, k - list.size()));
            }
        }
        int[] res = new int[k];
        for (int i = 0; i < k; i++) {
            res[i] = list.get(i);
        }
        return res;
    }
}