top-k-frequent-elements: use Counter

Author: haklee

top-k-frequent-elements/haklee.py

@@ -15,9 +15,9 @@
 리스트의 앞 k개의 아이템 리턴.
 """
 
+from collections import Counter
+
 
 class Solution:
     def topKFrequent(self, nums: List[int], n: int) -> List[int]:
-        d = {}
-        [d.update({i: d.get(i, 0) + 1}) for i in nums]
-        return [i for _, i in sorted([(-v, k) for k, v in d.items()])][:n]
+        return [i for _, i in sorted([(-v, k) for k, v in Counter(nums).items()])][:n]