leetcode347.前K个高频元素 Python

题目：

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

示例：

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

思路：

桶排序

先建立一个字典来统计数组中每个数出现的次数，然后建立一个二维数组，这个二维数组的索引表示每个桶的序号，初始化每个桶是一个空列表，然后读取字典中的键值对，让值也就是出现的频率作为二维数组的索引，将键也就是数放进对应的桶里，这样出现频率相同的数就放进了同一个桶里，并且出现频率就是对应的索引，即出现频率是由小到大排列的。

将桶整体逆序，然后来找前K个元素，当桶非空时，并且K大于0的情况下，开始往结果里加元素，直到当前桶内元素为空的时候，才有可能遍历下一个桶，前提是K大于0,每在结果里加一个元素，k减1,对结果中的元素计数，直到k等于0直接返回结果，得到频率前K高的元素数组。

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        dict1 = {}
        res = []
        for i in range(len(nums)):
            if nums[i] not in dict1:
                dict1[nums[i]] = 1
            else:
                dict1[nums[i]] += 1
        list1 = [[] for _ in range(len(nums) + 1)]
        for key in dict1:
            index = dict1[key]
            list1[index].append(key)
        list1.reverse()
        for item in list1:
            if item and k > 0:
                while item and k > 0:
                    res.append(item.pop())
                    k -= 1
            if k == 0:
                return res
        return res