使用python的dict存储,value记录数量,再排序,最后输出;
class Solution(object):
def topKFrequent(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
dic = dict()
for item in nums:
if item not in dic:
dic[item] = 1
else:
dic[item] += 1
dic = sorted(dic, key=dic.__getitem__, reverse = True)
print(dic)
res = []
for item in dic:
if len(res) < k:
res.append(item)
return res