Update sqrt_x.py

This commit is contained in:
marina 2023-08-07 21:49:19 -07:00 committed by GitHub
parent 855ecff683
commit e63cfe856f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -12,16 +12,16 @@ def sqrt(x) -> int:
while left <= right:
pivot = left + (right - left) // 2
num = pivot * pivot
mid = (right + left) // 2
num = mid * mid
if num > x:
right = pivot - 1
right = mid - 1
elif num < x:
left = pivot + 1
left = mid + 1
else:
return pivot
return mid
return right