《LeetCode力扣练习》代码随想录——栈与队列（前 K 个高频元素—Java）

刷题思路来源于 代码随想录

347. 前 K 个高频元素

• 堆

  class Solution {
      public int[] topKFrequent(int[] nums, int k) {
  
          if(nums.length==1){
              return nums;
          }
  
          Map<Integer,Integer> map=new HashMap<>();
  
          for(int num:nums){
              map.put(num,(map.getOrDefault(num,0)+1));
          }
  
          PriorityQueue<int[]> heap=new PriorityQueue<>((o1,o2)->o1[1]-o2[1]);
  
          for(Map.Entry<Integer,Integer> entry:map.entrySet()){
  
              if(heap.size()<k){
                  heap.offer(new int[]{entry.getKey(),entry.getValue()});
              }else{
  
                  if(entry.getValue()>heap.peek()[1]){
                      heap.poll();
                      heap.offer(new int[]{entry.getKey(),entry.getValue()});
                  }
  
              }
  
          }
  
          int[] result=new int[k];
  
          for(int i=k-1;i>=0;i--){
              result[i]=heap.poll()[0];
          }
  
          return result;
  
      }
  }