Method topKFrequent

Array/TopKFrequentElements.py:29–44

:type nums: List[int] :type k: int :rtype: List[int]

(self, nums, k)

"""
class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        
        nums_dict = {}
        
        for i in nums:
            try:
                nums_dict[i] += 1
            except KeyError:
                nums_dict[i] = 1
        
        return sorted(nums_dict, key=lambda x: nums_dict[x], reverse=True)[:k]