关于“前K个高频元素”的题解及复杂度分析

最新推荐文章于 2023-11-11 19:05:23 发布

玩技术因为喜欢

最新推荐文章于 2023-11-11 19:05:23 发布

阅读量213

点赞数

本文链接：https://blog.csdn.net/u013667527/article/details/127067649

版权

在做力扣347. 前 K 个高频元素题目时，发现用Python的Counter函数进行统计，然后对统计后的字典按照value进行排序，这也是个不错的方法，简单又直观。下面来分析一下复杂度，首先利用Counter函数对数组中元素的出现次数进行统计，这一步的时间复杂度时O(n)，这个不难理解，对频次进行统计，就是对数组遍历了一次。第二步，对统计后的字典（或者转化为二维数组）按照value值（频率）进行排序，这一步的时间复杂度为O((logn) * (log(logn)))，远低于题目“进阶”要求的复杂度，那么为什么复杂度为O((logn) * (log(logn)))呢？首先我们知道排序算法的默认时间复杂度为O(n * logn)，题目中明确说明‘数组中前 k 个高频元素的集合是唯一的’，这也就意味着题目中的k = logn，因为对字典排序的时间复杂度时O(k * logk)，因此最终的时间复杂度为O((logn) * (log(logn)))，执行结果显示程序运行时间优于大多数提交者。

下面给出具体代码，有时间的小伙伴可以优化一下代码。（欢迎评论、交流）

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        d = collections.Counter(nums)
        ret = []
        for ele in d.items():
            ret.append(ele)
        ret.sort(key = lambda x: x[1], reverse = True)
        ans = []
        for i in range(k):
            ans.append(ret[i][0])
        return ans