diff --git a/NeetCodeRoadmap/Arrays_and_Hashing/347_Top_K_Frequent_Elements.py b/NeetCodeRoadmap/Arrays_and_Hashing/347_Top_K_Frequent_Elements.py index ceae695..6df7356 100644 --- a/NeetCodeRoadmap/Arrays_and_Hashing/347_Top_K_Frequent_Elements.py +++ b/NeetCodeRoadmap/Arrays_and_Hashing/347_Top_K_Frequent_Elements.py @@ -5,7 +5,16 @@ def topKFrequent(nums, k): :rtype: List[int] """ frequency = {} + elements = [] for x in nums: - + if x not in frequency: + frequency[x] = 1 + else: + frequency[x] += 1 + for y in frequency: + if frequency[y] >= k: + elements.append(y) + return elements + print(topKFrequent([1,1,1,2,2,3], 2)) \ No newline at end of file