这个题目虽然不够新,但是还是需要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