leetcode 347

这个题目虽然不够新,但是还是需要mark一下,解决了我的一些误区。这题的想法很简单,就是需要统计一下数字出现的频率,然后排个序,输出前K个,就Ok了。但是要求复杂度小于nlgn。想了半天,觉得一定要排序呀,后来看了discuss才发现,自己忘了bucket sort。一个典型的牺牲空间换时间的算法。主要就是列出一个足够长的数组,把数字按照频率放在对应的位置上面。最后再从后向前遍历一遍,就得到结果了。

此外还有一点需要注意,我在初始化list的时候,用的是list = [[-1000]] * n 这样的形式,结果发现在对其中一个进行append的时候,其他item也全部都会改变。后来查阅文档就会发现,它应该是类型与C++指针的形式,每一项的地址都是一样的,这样在索引的时候自然都会得到一样的结果。所以我们需要采用copy这个操作,来开辟一块新的地址,再进行赋值,这样才能够顺利完成。附代码:

class Solution:
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        fre = {}
        for i in range(len(nums)):
            val = nums[i]
            if val in fre:
                fre[val] +=1
            else:
                fre[val] = 1
        bucket = [[-10000]] * (len(nums)+1)
        for key,value in fre.items():
            temp = bucket[value].copy()
            temp.append(key)
            bucket[value] = temp
        result = []
        for i in range(len(bucket)-1,-1,-1):
            for j in bucket[i]:
                if j != -10000:
                    result.append(j)
                    if len(result) >= k:
                        return result
        return result

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值