Create Solution3.go

Author: yanglbme

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

@@ -0,0 +1,29 @@
+func kClosest(points [][]int, k int) (ans [][]int) {
+	n := len(points)
+	dist := make([]int, n)
+	l, r := 0, 0
+	for i, p := range points {
+		dist[i] = p[0]*p[0] + p[1]*p[1]
+		r = max(r, dist[i])
+	}
+	for l < r {
+		mid := (l + r) >> 1
+		cnt := 0
+		for _, d := range dist {
+			if d <= mid {
+				cnt++
+			}
+		}
+		if cnt >= k {
+			r = mid
+		} else {
+			l = mid + 1
+		}
+	}
+	for i, p := range points {
+		if dist[i] <= l {
+			ans = append(ans, p)
+		}
+	}
+	return
+}