[99클럽] 31일차 TIL

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

 

정렬하여 카운트를 해주고, 많은 숫자를 보여한 숫자를 정답칸에 K만큼 저장해주고 반환하는 문제입니다.

 

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        cnt = {}
        cnt_list = [[] for _ in range(len(nums) + 1)]
        answer = []

        for num in nums:
            cnt[num] = 1 + cnt.get(num, 0)
        for key, val in cnt.items():
            cnt_list[val].append(key)
        
        for i in range(len(cnt_list) - 1, 0, -1):
            for j in cnt_list[i]:
                answer.append(j)
                if len(answer) == k:
                    return answer