LeetCode--347. Top K Frequent Elements（前K个高频元素）Python

诚实的小小乐

于 2018-01-02 16:10:25 发布

阅读量2.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/xiaoxiaoley/article/details/78952751

LeetCode 专栏收录该内容

101 篇文章

订阅专栏

本文介绍了一种求解数组中出现次数最多的前K个元素的方法。通过使用哈希表记录每个元素出现的次数，然后根据出现次数进行排序，最终找到频次最高的K个元素。

题目：

给定一个数组，求出该数组中出现次数最多的前K个元素内容。要求复杂度为 nlog(n) .

解题思路：

使用哈希表（python 中字典）来存储各个元素出现的次数，键（key）为对应元素，值（value）为对应元素出现个数。再对该哈希表（字典）按照值进行排序。

代码（Python）：

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        Dict = {}
        for i in range(len(nums)):
            if nums[i] in Dict:
                Dict[nums[i]] = Dict[nums[i]] + 1
            else:
                Dict[nums[i]] = 1
        
        output = sorted(Dict.items(),key=lambda e:e[1],reverse=True)
        
        final = []
        for i in range(k):
            final.append(output[i][0])
        return final