LeetCode347. 前 K 个高频元素（含详细解析JAVA实现）

题目描述

步骤分析

首先 用map来存储nums[]中各个数字出现的个数然后 创建也给优先队列并给它添加compare（）使存储队列内的元素按照key值从小到大排列把map中的值添加到优先队列中（只加入k个元素）使优先队列中的元素就是题目所需要的定义一个数组存放最终结果

参考代码

class Solution {
   
	  public int[] topKFrequent(int[] nums, int k) {
	 	 //用map来存储nums[]中各个数字出现的个数
		 Map map = new HashMap<>();
		 for(int num:nums) {
			 map.put(num, map.getOrDefault(num, 0)+1);
		 }
		 //创建也给优先队列并给它添加compare（）使存储队列内的元素按照key值从小到大排列
		 PriorityQueue queue = new PriorityQueue<>(new Comparator() {
			@Override
			public int compare(Integer o1, Integer o2) {
				 
				return map.get(o1)-map.get(o2);
			}
		});
		 
		 //把map中的值添加到优先队列中（只加入k个元素）使优先队列中的元素就是题目所需要的
		 for(Integer key: map.keySet()) {
			 if(queue.size()map.get(queue.peek())) {
				 queue.poll();
				 queue.add(key);
			 }
		 }
		 
		 //定义一个数组存放最终结果
		 int[] res =new int[k];
		 int index=0;
		 while(!queue.isEmpty()) {
			 res[index++]=queue.poll();
		 }
		return res;
	}
}