题目:
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842...,
由于返回类型是整数,小数部分将被舍去。
思路是二分法
我的最初答案:
class Solution:
def mySqrt(self, x):
"""
:type x: int
:rtype: int
"""
if x==0:
return 0
if x==1:
return 1
min = 0
max = x
mid = (min+max)//2
while max-min>1:
m = mid*mid
if m>x:
max = mid
mid = (min+max)//2
elif m<x:
min = mid
mid = (min+max)//2
else:
return mid
break
return min
我的这个运行时间比较长,原因可能是最开始判断了0和1的情况,而且没有很恰当的使用二分法。通过学习别人的代码,发现自己的不足,代码如下:
class Solution:
def mySqrt(self, x):
"""
:type x: int
:rtype: int
"""
min = 0
max = x
mid = x//2
while min<=max:
m = mid*mid
if m>x:
max = mid-1
elif m<x:
min = mid+1
else:
return mid
break
mid = (min+max)//2
return mid