347. Top K Frequent Elements

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.

 

本题有一个要求就是算法时间复杂度要好于nlongn，本题看了答案才知道使用bucket的方法做出来，首先，bucket是把问题分成数组长度+1的等份，然后将数组输入到hashmap里面，其中value值代表的是出现该值的频率，然后遍历hashmap的key值，将其放入到其value对应的bucket数组里面，代码如下：

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