can be further optimized on the increment

luiznasc · Oct 4, 2023 · 49f9a89 · 49f9a89
1 parent 8496593
commit 49f9a89
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/mySqrt/mySqrt.go b/mySqrt/mySqrt.go
@@ -1,13 +1,23 @@
 package main
 
 func mySqrt(x int) int {
-	count := 0
+	count := x
+	upperEdge := x
 	for {
 		if count*count > x {
-			count--
-			break
+			upperEdge = count
+			count = count / 2
+		} else {
+			if count*count == x {
+				return count
+			}
+			for i := count; i <= upperEdge; i++ {
+				if i*i > x {
+					return i - 1
+				} else if i*i == x {
+					return i
+				}
+			}
 		}
-		count++
 	}
-	return count
 }