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

LeetCode347. 前 K 个高频元素

题目描述

步骤分析

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

参考代码

class Solution {
   
	  public int[] topKFrequent(int[] nums, int k) {
	 	 //用map来存储nums[]中各个数字出现的个数
		 Map<Integer,Integer> map = new HashMap<>();
		 for(int num:nums) {
			 map.put(num, map.getOrDefault(num, 0)+1);
		 }
		 //创建也给优先队列并给它添加compare（）使存储队列内的元素按照key值从小到大排列
		 PriorityQueue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>() {
			@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()<k) {
				 queue.add(key);
			 }else if(map.get(key)>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;
	}
}