实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842...,
由于返回类型是整数,小数部分将被舍去。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sqrtx
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
最开始我想到的是一种比较野蛮得方法去求解,但是这种方法在验证的时候直接提示超时了:
class Solution:
def mySqrt(self, x: int) -> int:
if x <= 1:
return x
half = int(x//2) + 1
res = -1
for i in range(half):
if i * i <= x:
res = i
return res
二分法求解:
class Solution:
def mySqrt(self, x: int) -> int:
if x <= 1:
return x
left, right = 0, x
res = -1
while left <= right:
mid = left + (right - left)//2
print("mid:%d left:%d right:%d"%(mid, left, right))
if mid * mid <= x:
res = mid
left = mid + 1
else:
right = mid - 1
print("mid:%d left:%d right:%d res:%d"%(mid, left, right, res))
return int(res)