Leetcode刷题之——前K个高频元素（中等难度）-python解决

小小数据挖掘工程师

于 2020-02-01 19:51:15 发布

阅读量546

点赞数

分类专栏： LeetCode 编程练习

本文链接：https://blog.csdn.net/lu_yunjie/article/details/104137613

版权

编程练习同时被 2 个专栏收录

269 篇文章

订阅专栏

LeetCode

50 篇文章

订阅专栏

本文介绍了如何使用Python解决LeetCode中等难度问题——找出数组中出现频率前K的元素。提供了两种解决方案，一是利用计数函数，二是应用桶排序算法，并附有详细的代码实现及提交结果。

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

题目描述：

解决方案一：计数函数直接返回

代码如下:

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        return [item[0] for item in collections.Counter(nums).most_common(k)]

提交结果：

解决方案二：桶排序

代码如下：

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        from collections import Counter
        
        c = Counter(nums)
        buckets = [[] for _ in  range(len(nums) + 1)]
        
        for x, y in c.items():
            buckets[y].append(x)
        res = [] 
        for i in range(len(nums), -1, -1):
            if len(res) > k:
                break
            res.extend(buckets[i])
        return res[:k]

提交结果：