(Leetcode) 前 K 个高频元素 - Python实现

题目：前 K 个高频元素
给定一个非空的整数数组，返回其中出现频率前 k 高的元素。
示例:
输入: nums = [1,1,1,2,2,3], k = 2。输出: [1,2]
输入: nums = [1], k = 1。输出: [1]
说明:
你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

------------------------------------------------------------------------------------------------

解法1：通过字典排序的方式，时间复杂度 O(n log n)

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        if len(nums) == 0:
            return []

        dic = {}
        for num in nums:
            if num not in dic.keys():
                dic[num] = 1
            else:
                dic[num] += 1

        li = sorted(dic.items(), key=lambda x:x[1], reverse=True)
        return [item[0] for item in li[:k]]

解法2#: 通过Counter函数

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        if len(nums) == 0:
            return []

        li = collections.Counter(nums)
        return [x[0] for x in li.most_common(k)]

解法3：通过heapq方法

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """ 
        if len(nums) == 0:
            return []

        import heapq
        dic = dict()
        for num in nums:
            dic[num] = dic.get(num, 0) + 1

        li = list()
        for item in dic.items():
            if len(li) == k:
                if item[1] > li[0][0]:
                    heapq.heappop(li)
                    heapq.heappush(li, (item[1], item[0]))
            else:
                heapq.heappush(li, (item[1], item[0]))
        return [item[1] for item in li]

参考：

https://www.cnblogs.com/MartinLwx/p/9707751.html

https://blog.csdn.net/ustbbsy/article/details/79637594