Leetcode347. 前K个高频元素

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

例如，

给定数组 [1,1,1,2,2,3] , 和 k = 2，返回 [1,2]。

注意：

• 你可以假设给定的 k 总是合理的，1 ≤ k ≤ 数组中不相同的元素的个数。
• 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
•               (1) 使用字典Map 和 优先队列PriorityQueue
•                // Map<Integer,Integer> map = new HashMap<>();
  // for(int e:nums) {
  // map.put(e, map.getOrDefault(e, 0)+1);
  // }
  // PriorityQueue<Map.Entry<Integer, Integer>> pq = new PriorityQueue<>(
  // (o1,o2)->o1.getValue()-o2.getValue());
  // for(Map.Entry<Integer, Integer> entry:map.entrySet()) {
  // if(pq.size()<k)
  // pq.offer(entry);
  // else if(pq.peek().getValue()<entry.getValue()) {
  // pq.poll();
  // pq.offer(entry);
  // }
  // }
  // List<Integer> list = new ArrayList<>();
  // for(Map.Entry<Integer, Integer> e:pq) {
  // list.add(e.getKey());
  // }
  // return list;
• (2) 使用桶排序，每一个桶的下标代表出现的次数，桶内装出现同样多的数
          // use an array to save numbers into different bucket whose index is the frequency
          Map<Integer, Integer> map = new HashMap<>();
          for(int n: nums){
              map.put(n, map.getOrDefault(n,0)+1);
          }
          
          // corner case: if there is only one number in nums, we need the bucket has index 1.
          List<Integer>[] bucket = new List[nums.length+1];
          for(int n:map.keySet()){
              int freq = map.get(n);
              if(bucket[freq]==null)
                  bucket[freq] = new LinkedList<>();
              bucket[freq].add(n);
          }
          
          List<Integer> res = new LinkedList<>();
          for(int i=bucket.length-1; i>0 && k>0; --i){
              if(bucket[i]!=null){
                  List<Integer> list = bucket[i]; 
                  res.addAll(list);
                  k-= list.size();
              }
          }
          
          return res;
      }