【leetcode】 347. 前 K 个高频元素

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]
说明：

你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

统计每个数的频率，输出最大的几个，这完全迎合了Python中的Counter类，调用其的几个方法即可。

什么是Counter？

Counter 是一个在collections包里的类，正如其名，是一个用于计数的工具。
我们可以用Counter(nums)这样的构造函数构造一个Counter类，其中nums是一个列表。
构造好的Counter实例可以看作一个字典，键是nums的每一项，值是它的出现次数。
如果上面的叙述让你感到很混乱的话，我不妨举个例子。
如果一个列表a = [1,1,3,4,3]，你想要统计每项的出现次数，那么你使用b = Counter(a)，那么这时候b就像一个这样的字典{1:2,3:2,4:1}，表示数字1出现了2次，数字3出现了2次，数字4出现了1次。
还是很好理解的吧？
可是题目里要我们输出的是最多的K项
这时候可以应用Counter的一个函数，most_common(k)
这个函数就是返回最多出现的K项
但是返回的形式是一个元祖列表，类似[(1,2),(3,2),(4,1)]的形式
我们只需要键也就是第一项，所以要使用列表生成式处理一下即可。

代码

from collections import Counter
class Solution:
def topKFrequent(self, nums: List[int], k: int) -> List[int]:
return [i[0] for i in Counter(nums).most_common(k)]

作者：jimmy00745
链接：https://leetcode-cn.com/problems/top-k-frequent-elements/solution/yi-xing-python3dai-ni-zou-jin-counterlei-by-jimmy0/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。