leetcode347. 前K个高频元素(Hashmap+桶排序)

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]

先用HashMap记录频率,再用 list数组中每一个位置为这个数出现的次数 值为出现次数这么多的数,再从后往前遍历

class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
         List<Integer> res = new ArrayList();
   
        HashMap<Integer,Integer> map = new HashMap();

        for(int i=0;i<nums.length;i++){
            if(map.containsKey(nums[i])){
                map.put(nums[i],map.get(nums[i])+1);
            }
            else{
                map.put(nums[i],1);
            }      
        }
        
  List<Integer>[] list = new List[nums.length+1];
     
        for(int key : map.keySet()){
            int i = map.get(key);
     if(list[i]==null){
       list[i] = new ArrayList();  
     }
        list[i].add(key) ;    
        }
for(int i=nums.length;i>=0&&res.size()<k;i--){
       if(list[i]==null) continue;
            res.addAll(list[i]);
}
       return res; 
    }
}

你可能感兴趣的:(mid)