Posted on 2023-05-22 16:20
Uriel 閱讀(51)
評論(0) 編輯 收藏 引用 所屬分類:
閑來無事重切Leet Code 、
大水題
給出一個數組，輸出出現頻次最高的k個數（輸出的順序無限制），dict統計頻次再sort
1 #347
2 #Runtime: 76 ms (Beats 90.9%)
3 #Memory: 16.8 MB (Beats 53.94%)
4
5 class Solution(object):
6 def topKFrequent(self, nums, k):
7 """
8 :type nums: List[int]
9 :type k: int
10 :rtype: List[int]
11 """
12 cnt = defaultdict(lambda:0)
13 for i in nums:
14 cnt[i] += 1
15 a = sorted(cnt.items(), key=lambda item: item[1], reverse=True)
16 ans = []
17 for x in a:
18 ans.append(x[0])
19 return ans[0:k]