최종 종합퀴즈 — 전 챕터 복합 문제

from collections import Counter
import heapq

def top_k_frequent(nums, k):
    # 1. 빈도 계산 — O(n)
    count = Counter(nums)
    
    # 2. 힙으로 Top K 추출 — O(n log k)
    return heapq.nlargest(k, count.keys(), key=count.get)

# 또 다른 접근: Bucket Sort — O(n)
def top_k_frequent_bucket(nums, k):
    count = Counter(nums)
    
    # 빈도별 버킷 (인덱스 = 빈도)
    buckets = [[] for _ in range(len(nums) + 1)]
    for num, freq in count.items():
        buckets[freq].append(num)
    
    # 높은 빈도부터 k개 수집
    result = []
    for freq in range(len(buckets) - 1, 0, -1):
        for num in buckets[freq]:
            result.append(num)
            if len(result) == k:
                return result
    return result