给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2 输出: [1,2]
示例 2:
输入: nums = [1], k = 1 输出: [1]
思路:统计每个元素出现次数,然后排序,最后输出前k个。很多人会想用字典统计次数,但字典是无序的,你还得先对字典按值进行排序,python提供了一个Counter函数,他会统计一个列表中各个元素出现的次数,并返回一个元组,然后利用most_common函数,它会按照出现频率降序返回一个元组,如果不给参数,那就全部返回,若给了参数k,那就返回出现次数最多的前k个。
这么做是利用了python中封装好的函数,属于是投机取巧了,如果不知道这两个函数压根做不出来,但这道题别的解法是利用大顶堆/小顶堆,暂时还没看懂,先放着吧,后面再来更新
代码(Python):
class Solution(object):
def topKFrequent(self, nums, k):
count = collections.Counter(nums)
return [num[0] for num in count.most_common(k)]