【优先队列】347. 前 K 个高频元素

力扣链接：. - 力扣（LeetCode）

思路：用优先队列的直接按出现频率排序

时间复杂度：O(Nlogk)，其中 N 为数组的长度。我们首先遍历原数组，并使用哈希表记录出现次数，每个元素需要 O(1) 的时间，共需 O(N) 的时间。随后，我们遍历「出现次数数组」，由于堆的大小至多为 k，因此每次堆操作需要 O(logk) 的时间，共需 O(Nlogk) 的时间。二者之和为 O(Nlogk)

空间复杂度：O(N)。哈希表的大小为 O(N)，而堆的大小为 O(k)，共计为 O(N)

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> mp = new HashMap<>();
        for (int i : nums) {
            int num = mp.getOrDefault(i, 0) + 1;
            mp.put(i, num);
        }
        PriorityQueue<Integer> que = new PriorityQueue<>((s1, s2) -> mp.get(s2) - mp.get(s1));
        for (int num : mp.keySet()) {
            que.add(num);
        }
        int[] ans = new int[k];
        for (int i = 0; i < k; i++) {
            ans[i] = que.poll();
        }
        return ans;

    }
}