347. Top K Frequent Elements [medium] (Python)

题目链接

https://leetcode.com/problems/top-k-frequent-elements/

题目原文

Given a non-empty array of integers, return the k most frequent elements.

For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].

Note:

1. You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
2. Your algorithm’s time complexity must be better than O(n log n), where n is the array’s size.

思路方法

解这个题的关键在于控制时间复杂度“小于O(nlogn)”这个条件。

思路一

按照我的思维习惯，看到这个题，首先想到的是下面的思路：先用dict得到所有不同数的个数；再对个数排序，取前k个个数最多的对应的数即可。代码如下，然而使用了内置sorted()函数，只能说时间复杂度小于等于O(nlogn)，不太满足题意的样子，仅供参考。【时间复杂度O(nlogn)】

代码

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        data, re