Create Solution3.py

Author: yanglbme

solution/0900-0999/0973.K Closest Points to Origin/Solution3.py

@@ -0,0 +1,12 @@
+class Solution:
+    def kClosest(self, points: List[List[int]], k: int) -> List[List[int]]:
+        dist = [x * x + y * y for x, y in points]
+        l, r = 0, max(dist)
+        while l < r:
+            mid = (l + r) >> 1
+            cnt = sum(d <= mid for d in dist)
+            if cnt >= k:
+                r = mid
+            else:
+                l = mid + 1
+        return [points[i] for i, d in enumerate(dist) if d <= l]