[LeetCode][Python]Top K Frequent Elements

op 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: 

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

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

---

求出现频率最高的K个数，要求时间复杂度不能超过 O(n log n)。

选择最简单的快排O(n log n)来解决，先求出每个数出现的几率，然后排序求出结果。

排序的地方用lambda函数，就是一个匿名方法，如果是升序，写成：

list.sort(key=lambda x:x[1])

相当于:

list.sort(lambda a,b:cmp(a[1],b[1]))

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        self.__dict = {}; list = []; res = []
        for item in nums:
            if self.__dict.has_key(item):
                self.__dict[item] += 1
            else:
                self.__dict[item] = 1
        for item in self.__dict:
            list.append((item, self.__dict[item]))
        list.sort(key=lambda x:-x[1])
        for i in range(k):
            res.append(list[i][0])
        return res