剑指 Offer（专项突击版）- 剑指 Offer II 060. 出现频率最高的 k 个数字(20220807)

给定一个整数数组 nums 和一个整数 k ，请返回其中出现频率前 k 高的元素。可以按 任意顺序 返回答案。

 

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]
 

提示：

1 <= nums.length <= 105
k 的取值范围是 [1, 数组中不相同的元素的个数]
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的
 

进阶：所设计算法的时间复杂度 必须 优于 O(n log n) ，其中 n 是数组大小。


class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        numsfreqsdcit = collections.Counter(nums)
        minHeap = []

        for num, freq in numsfreqsdcit.items():
            if len(minHeap) == k:
                if minHeap[0][0] < freq:
                    heapq.heappop(minHeap)
                    heapq.heappush(minHeap, (freq, num))
            else:
                heapq.heappush(minHeap, (freq, num))

        print(minHeap)
        res = []
        while minHeap:
            freq, num = minHeap.pop()
            res.append(num)

        return res