leetcode热题100.前k个高频元素

作者：晓宜
🌈🌈🌈
个人简介：互联网大厂Java准入职，阿里云专家博主，csdn后端优质创作者，算法爱好者
❤️❤️❤️
你的关注是我前进的动力😊

Problem: 347. 前 K 个高频元素

题目

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

示例 1:

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

输出: [1,2]

示例 2:

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

提示：

• $1<=nums.length<=10^5$
• k 的取值范围是 [1, 数组中不相同的元素的个数]
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

进阶：你所设计算法的时间复杂度 必须 优于 O(n log n) ，其中 n 是数组大小。

思路

首先统计每个元素出现的次数

然后建立一个包含k个元素的最小堆，我们先用循环构建这样一个堆，

如果此时nums中还有元素的话，我们把这个元素和堆顶元素的出现频率做比较，如果当前元素的频率更高，说明我们应该弹出堆中出现频率最小的元素，然后加入这个元素

最后，这个堆的元素就是nums中出现频率前k高的元素了，我们把他们的key重新构造成数组返回

Code

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        h = []
        d = defaultdict(int)
        for x in nums:
            d[x] += 1
        
        
        for key,v in d.items():
            if len(h)<k:
                heappush(h,(v,key))
            else:
                if v>h[0][0]:
                    heappop(h)
                    heappush(h,(v,key))
        
        return [key for v,key in h]