動機

直接sort就好

Problem

Given an integer array nums and an integer k, return the k most frequent elements. You may return the answer in any order.

 

Example 1:

Input: nums = [1,1,1,2,2,3], k = 2Output: [1,2]

Example 2:

Input: nums = [1], k = 1Output: [1]

 

Constraints:

  • 1 <= nums.length <= 105
  • k is in the range [1, the number of unique elements in the array].
  • It is guaranteed that the answer is unique.

 

Follow up: Your algorithm's time complexity must be better than O(n log n), where n is the array's size.

Sol

sort就是O(nlog(n)),也不必要用heap,因為不用一直塞資料到heap中,所以直接sort就好

class Solution:
    def topKFrequent(self, nums: List[int], b: int) -> List[int]:
        tmp = sorted([(k,v) for (k,v) in Counter(nums).items()], key=lambda x: x[1], reverse=True)
        return [k for (k,v) in tmp[:b]]