347. Top K Frequent Elements
-
347. Top K Frequent ElementsProgramming/leetcode 2021. 2. 22. 21:02
주어진 리스트에서 빈도수를 세어놓고, k순위만큼 출력하는 문제 collection.Counters를 사용하면 쉽게 풀린다. def topKFrequent(self, nums: [], k: int) -> []: lst = collections.Counter(nums).most_common(k) return [l[0] for l in lst] 책에서는 heap을 사용해서 했던데, 방법이 신기했다. def topKFrequent(self, nums: [], k: int) -> []: freqs = collections.Counter(nums) freqs_heap = [] for f in freqs: heapq.heappush(freqs_heap, (-freqs[f], f)) print(freqs_heap) t..