LeetCode 347. Top K Frequent Elements - 优先级队列（Priority Queue）系列题3

原创于 2022-02-03 06:06:46 发布 · 298 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数组结构 #python #优先级队列

Leetcode刷题笔记同时被 2 个专栏收录

198 篇文章

订阅专栏

优先级队列

12 篇文章

订阅专栏

该博客介绍了如何利用Python的优先级队列（最小堆）解决LeetCode上的347题，即在给定整数数组nums和整数k的情况下，找出出现频率最高的k个元素。博主通过遍历数组，统计每个数字的频率，并将频率作为优先级放入最小堆中，保证了高频元素优先。当堆大小超过k时，移除频率最低的元素。最终堆中的k个元素即为所求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an integer array nums and an integer k, return the k most frequent elements. You may return the answer in any order.

Example 1:

Input: nums = [1,1,1,2,2,3], k = 2
Output: [1,2]

Example 2:

Input: nums = [1], k = 1
Output: [1]

Constraints:

1 <= nums.length <= 105
k is in the range [1, the number of unique elements in the array].
It is guaranteed that the answer is unique.

题目给定一个数组nums和一个整数k，有些数字在数组nums里可能出现多次。要求找出出现频率最高的k个数字。

很明显是一道Top K的题目，可以用优先级队列来搞定，由于要保留频率高的数字排除频率低的数字，可以使用最小堆，heapq默认就是最小堆：用Python刷LeetCode必备知识点1 - 优先级队列。首先遍历一次数组，用一个dict来记录每个数字出现的次数。然后遍历dict把数字及其频率作为一个元组tuple（频率在前数字在后）放入优先级队列，一旦队列中个数超过k就弹出一个，由于是最小堆，被弹出的肯定是队列里频率最小的数字。最后队列里的k个数字就是出现频率最高的。

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        freq = {}
        
        for num in nums:
            if num in freq:
                freq[num] += 1
            else:
                freq[num] = 1
        
        pq = []
        for num,cnt in freq.items():
            heapq.heappush(pq, (cnt, num))
            if len(pq) > k:
                heapq.heappop(pq)
        
        return [num for (_, num) in pq]