leetcode hot100 347. 前 K 个高频元素 medium 桶排序

---

频率前 k 高的元素

时间复杂度：O(n)

空间复杂度：O(n)

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        
        # 1.用哈希表统计频率
        count = Counter(nums)
        #print(count.items())   # dict_items([(1, 3), (2, 2), (3, 1)])  (num, freqs)

        #2. 创建一个长度为 n+1 的空数组（桶数组）
        n = len(nums)
        buckets = [[]  for _ in range(n+1)]

        # 索引= 出现频率
        for num, freqs in count.items():
            buckets[freqs].append(num)   # 向每个频率的空列表中，加入对应的num值

        #注意： 每个桶里可能有多个数字，buckets[2] = [1, 2] → 频率为 2 的数字有两个，分别是1和2
        
        # 从最高频开始，把元素加入res，直到加入k个（频率前 k 高的元素）
        res = []
        for freq in range(n, 0, -1):  # 从最高频开始，num加入res
            for num in buckets[freq]:
                res.append(num)

                if len(res) == k:
                    return res