70-Heap--LC692前K个高频单词

原创已于 2022-03-18 21:12:41 修改 · 188 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

于 2022-03-18 20:53:47 首次发布

leetcode 专栏收录该内容

75 篇文章

订阅专栏

该博客探讨了如何使用Python的heapq模块和collections.Counter来有效地找到列表中出现频率最高的K个元素。首先，通过Counter计算每个元素的频率，然后利用堆数据结构将元素及其频率转换为堆。通过heappop操作，可以按频率降序获取Top K的元素。此外，还提到了使用排序实现相同功能的方法，但堆在处理大数据集时更具优势。

在这里插入图片描述
通常在问题中看到前 K 大，前 KK小或者第 K 个， K 个最等等类似字样,一般情况下我们都可以用堆来做。

class Solution(object):
    def topKFrequent(self, words, k):
        """
        :type words: List[str]
        :type k: int
        :rtype: List[str]
        """
        # collections.Counter(words)为python自带的计数器
        count = collections.Counter(words)
        heap = [(-freq, word) for word, freq in count.items()]
        heapq.heapify(heap)
        return [heapq.heappop(heap)[1] for _ in range(k)]

		# 普通排序
		d = collections.Counter(words)
        pairs = [(value, key) for key, value in d.items()]
        res = sorted(pairs, key = lambda a:(-a[0], a[1]))
        return [res[i][1] for i in range(k)]