【leetcode】347. Top K Frequent Elements

347. Top K Frequent Elements

https://leetcode.com/problems/top-k-frequent-elements/?tab=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.

简单来说，给一个不为空的整数数组，返回K个出现最多的数字。
这里如果要时间复杂度为O(n)，可以使用桶排序(bucket sort)

桶排序

一个简单的例子是桶式排序（bucket sort）。为使桶式排序能够正常工作，必须要有一些附加的信息。输入数据A1，A2，A3，…，AN必须只由小于M的正整数组成（显然还可以对其进行扩充）。如果是这种情况，那么算法很简单：使用一个大小为M的称为count的数组，它被初始化为全0。于是，count有M个单元或称桶，这些桶初始化为空。当读Ai时，count[Ai]增1。在所有的输入数据读入后，扫描数组count，打印出排序后的表。该算法用时O(M+N)。如果M为O(N)，那么总量就是O(N)。
http://www.linuxidc.com/Linux/2016-09/135312.htm

代码如下

public List<Integer> topKFrequent(int[] nums, int k) {

    List<Integer>[] bucket = new List[nums.length + 1];
    Map<Integer, Integer> frequencyMap = new HashMap<Integer, Integer>();

    for (int n : nums) {
        frequencyMap.put(n, frequencyMap.getOrDefault(n, 0) + 1);
    }

    for (int key : frequencyMap.keySet()) {
        int frequency = frequencyMap.get(key);
        if (bucket[frequency] == null) {
            bucket[frequency] = new ArrayList<>();
        }
        bucket[frequency].add(key);
    }

    List<Integer> res = new ArrayList<>();

    for (int pos = bucket.length - 1; pos >= 0 && res.size() < k; pos--) {
        if (bucket[pos] != null) {
            res.addAll(bucket[pos]);
        }
    }
    return res;
}