leetcode 347 号算法题：前 K 个高频元素

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。
你可以按 任意顺序 返回答案。输入: nums = [1,1,1,2,2,3], k = 2
输出: [2, 1]输入: nums = [1], k = 1
输出: [1]1 <= nums.length <= 10^5
k 的取值范围是 [1, 数组中不相同的元素的个数]
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的进阶：你所设计算法的时间复杂度 必须 优于 O(n log n) ，其中 n 是数组大小。

import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;public class Problem_347_TopKFrequentElements {public int [] topKFrequent(int[] nums, int k) {Map<Integer,Integer> count = new HashMap<>();for (int num : nums) {count.put(num, count.getOrDefault(num,0) + 1 );}PriorityQueue<Integer> pq= new PriorityQueue<>(k + 1,(a,b) -> count.get(a) - count.get(b));for (int num : count.keySet()) {pq.add(num);if(pq.size() > k) pq.remove();}int [] res = new int[k];int index = 0;while (!pq.isEmpty()){res[index++] = pq.remove();}return res;}
}