347. Top K Frequent Elements java solutions

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.

Subscribe to see which companies asked this question

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

最初的想法是 使用hashmap 记录频率，之后再按频率大小逆序输出k个元素。之后参考 discuss 中的代码，使用bucket来降低时间复杂度达到O(n)。