Python前k个高频的数字。

高频元素查找算法

最新推荐文章于 2025-06-04 15:30:40 发布

原创最新推荐文章于 2025-06-04 15:30:40 发布 · 561 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

Python笔记专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种有效的算法，用于从整数数组中找出出现频率最高的前k个元素。通过使用Python的Counter模块统计元素频率，并结合排序算法，该方法能够快速定位到目标元素，为数据处理和分析提供了一种实用的解决方案。

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

from collections import Counter
from typing import List


class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        dic2 = dict(Counter(nums))
        sort = sorted(dic2.items(), key=lambda e: e[1], reverse=True)
	
        res = []
        for i in range(k):
            res.append(sort[i][0])
        return res 

if __name__ == "__main__":
    s = Solution()
    list2 = [-1,-1]
    k =1 #
    print(s.topKFrequent(list2, k))