LeetCode 347. 前K个高频元素

题目

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

示例 1:

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

示例 2:

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

说明：

• 你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
• 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

题解

我自己的代码思路主要是先用一个字典记录 词-频 对，然后用开链法反向存成一个哈希表，同时将频数存成一个list, 从高到低排个序，挨个去反向的那个哈希表里找，如果同频的就只找一次，最后输出，时间复杂度方面应该是nlogn，主要是排序复杂度。代码如下：

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        result = []
        values = []
        reversehashtable = {}
        hashtable = {}
        for i in range(len(nums)):
            hashtable[nums[i]] = hashtable.get(nums[i],0) + 1
        for key in hashtable:
            reversehashtable[hashtable[key]] = reversehashtable.get(hashtable[key],[])
            reversehashtable[hashtable[key]].append(key)
            values.append(hashtable[key])
        print(values)
        values.sort(reverse = True)
        #print(hashtable, reversehashtable)
        for i in range(k):
            if i == 0:
                result.extend(reversehashtable[values[i]])
            if values[i] != values[i-1] and i > 0 :
                result.extend(reversehashtable[values[i]])
            if values[i] == values[i-1] and i > 0:
                continue
        return result

讨论区

一个想法是建立完哈希表之后用堆来实现前k个，所以堆的复杂度是nlogn