Merge pull request #2977 from coopers/0973

Update 0973-k-closest-points-to-origin.py

Author: sujal-goswami

python/0973-k-closest-points-to-origin.py

@@ -1,13 +1,13 @@
 class Solution:
     def kClosest(self, points: List[List[int]], k: int) -> List[List[int]]:
-        pts = []
+        minHeap = []
         for x, y in points:
-            dist = (abs(x - 0) ** 2) + (abs(y - 0) ** 2)
-            pts.append([dist, x, y])
-
+            dist = (x ** 2) + (y ** 2)
+            minHeap.append((dist, x, y))
+        
+        heapq.heapify(minHeap)
         res = []
-        heapq.heapify(pts)
-        for i in range(k):
-            dist, x, y = heapq.heappop(pts)
-            res.append([x, y])
-        return res
+        for _ in range(k):
+            _, x, y = heapq.heappop(minHeap)
+            res.append((x, y))
+        return res