347. Top K Frequent Elements

Partho Biswas · Partho Biswas · commit 0557bb6e3b43 · 2019-11-10T13:50:09.000+06:00
diff --git a/README.md b/README.md
@@ -222,7 +222,8 @@ I have solved quite a number of problems from several topics. See the below tabl
 ### [Heaps](https://www.youtube.com/watch?v=HqPJF2L5h9U)
 | #  | Title | Solution | Tutorial | Level | Remarks |
 | --- | --- | --- | --- | --- | --- |
-|01| [253_Meeting_Rooms_II](https://leetcode.com/problems/meeting-rooms-ii//)| [Python](leetcode.com/python/253_Meeting_Rooms_II.py)| [Official](https://leetcode.com/problems/meeting-rooms-ii/solution/) | Medium | --- |
+|01| [253_Meeting_Rooms_II](https://leetcode.com/problems/meeting-rooms-ii/)| [Python](leetcode.com/python/253_Meeting_Rooms_II.py)| [Official](https://leetcode.com/problems/meeting-rooms-ii/solution/) | Medium | --- |
+|02| [347. Top K Frequent Elements](https://leetcode.com/problems/top-k-frequent-elements/solution/)| [Python](leetcode.com/python/347_Top_K_Frequent_Elements.py)| [Official](https://leetcode.com/problems/meeting-rooms-ii/solution/) | Medium | --- |
 
 
 ### [Tries](https://leetcode.com/explore/learn/card/trie/)
diff --git a/leetcode.com/python/347_Top_K_Frequent_Elements.py b/leetcode.com/python/347_Top_K_Frequent_Elements.py
@@ -0,0 +1,37 @@
+from collections import Counter
+import heapq
+
+class Solution(object):
+
+    # # Using only HashMap/Dictionary
+    # def topKFrequent(self, nums, k):
+    #     """
+    #     :type nums: List[int]
+    #     :type k: int
+    #     :rtype: List[int]
+    #     """
+    #     count = Counter(nums)
+    #     elements = count.most_common(k)
+    #     common_keys = [x[0] for x in elements]
+    #     return common_keys
+
+
+    # Using heap and map
+    def topKFrequent(self, nums, k):
+        """
+        :type nums: List[int]
+        :type k: int
+        :rtype: List[int]
+        """
+        count = Counter(nums)
+        keys = count.keys()
+        common_keys = heapq.nlargest(k, keys, key=count.get) # key=count.get is used to sory y value in asending order
+        return common_keys
+
+
+
+sol = Solution()
+nums = [4,1,-1,2,-1,2,3]
+k = 2
+out = sol.topKFrequent(nums, k)
+print('Res: ',out)
