[LeetCode/Python] 347. 상위 K 빈도 요소(Top K Frequent Elements)

상위 K번 이상 등장하는 요소를 출력하라.

 

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

 

Output: [1,2]

 

 

 

 

1. 내가 푼 코드

 

 

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        freq_cnts = collections.Counter(nums)
        a = freq_cnts.most_common(k)
        b = []
        for i in range(k):
            b.append(a[i][0])

 

해설:

 

Counter를 이용하여 nums의 빈도수를 계산하고

 

most_common 함수를 이용하여 k개의 최빈값을 구한다.

 

 

 

 

 

 

 

** 깔끔한 답안 **

 

 

 

 

2. Counter를 이용한 음수 순 추출

 

 

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        freqs = collections.Counter(nums)
        freqs_heap = []
        # 힙에 음수로 삽입
        for f in freqs:
            heapq.heappush(freqs_heap, (-freqs[f], f))
 
        topk = list()
        # k번 만큼 추출, 최소 힙(Min Heap)이므로 가장 작은 음수 순으로 추출
        for _ in rnage(k):
            topk.append(heapq.heappop(freqs_heap)[1])
 
        return topk

 

해설:

 

Counter를 쓴 건 똑같은데 여기선 heapq를 이용하였고

 

heapq 모듈은 최소 힙(Min-Heap)만 지원하기에 빈도수를 음수로 바꿔 넣었다.

 

 

 

 

 

 

3. 파이썬다운 방식

 

 

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        return list(zip(*collections.Counter(nums).most_common(k)))[0]

 

해설:

 

파이썬에 내장된 zip과 *을 이용한 방식이다.

 

zip과 *의 쓰임은 별도로 설명하진 않겠다.

 

궁금하다면 아래 링크를 달아 두겠다.