出现频率最高的 k 个数字或前 K 个高频元素

题目

LeetCode
剑指 Offer

思路

• 使用HashMap存储数字到出现频率的映射。
• 使用大小为K的最小堆存储HashMap中的映射关系，使用PriorityQueue实现最小堆。建堆的规则采用出现频率作为依据。
• 最后将堆中的键值对数据中的键取出。

代码

import java.util.Comparator;
import java.util.HashMap;
import java.util.PriorityQueue;
import java.util.Map;

public class Solution {
	public int[] topKFrequent(int[] nums, int k) {
		HashMap<Integer, Integer> numToCount = new HashMap<>();
		// 统计频率
		for (Integer num : nums) {
			// 数字为键key,频率为实值value
			numToCount.put(num, numToCount.getOrDefault(num, 0) + 1);
		}

		// 创建用Map中的键值对key-value的实值value进行逆序排序的最小堆
		PriorityQueue<Map.Entry<Integer, Integer>> minHeap = new PriorityQueue<>(
				new Comparator<Map.Entry<Integer, Integer>>() {
					@Override
					public int compare(Map.Entry<Integer, Integer> e1, Map.Entry<Integer, Integer> e2) {
						// 大顶堆
						// return -(e1.getValue() - e2.getValue());
						// 小顶堆
						return e1.getValue() - e2.getValue();
					}
				});

		// 遍历映射，添加键值对到最小堆
		for (Map.Entry<Integer, Integer> entry : numToCount.entrySet()) {
			// 保持堆中最多只有K个元素
			if (minHeap.size() < k) {
				minHeap.offer(entry);
			} else if (entry.getValue() > minHeap.peek().getValue()) { // 大于堆顶
				minHeap.poll();
				minHeap.offer(entry);
			}
		}

		int[] res = new int[k];
		int i = 0;
		for (Map.Entry<Integer, Integer> entry : minHeap) {
			res[i++] = entry.getKey();
		}

		return res;
	}
}

• 后记：
  • 使用HashMap要熟练插入、遍历、获取键值、获取实值；
  • 理解Map.Entry接口，及其HashMap的内部类Entry；
  • Java中堆的实现PriorityQueue的常用方法及其创建。