给你一个非负整数 x
,计算并返回 x
的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5)
或者 x ** 0.5
。
示例 1:
输入:x = 4 输出:2
示例 2:
输入:x = 8 输出:2 解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
提示:
0 <= x <= 231 - 1
思路1:
用while循环来实现,i从0开始,直至 i * i > 我们想要输入的值(x),跳出循环,说明此时的 i 的前一个值即(i-1) 为该值 x 的算术平方根。
代码:
class Solution(object):
def mySqrt(self, x):
i=0
while(i*i<=x):
i=i+1
return i-1
复杂度分析:
a.时间复杂度:O(x^1/2)
b.空间复杂度:O(1)
思路二:二分查找
让左(l)为0,右(r)为x,一直找中间的数,如果中间的数(m)的平方大于x,说明此数大,右边就从中间减一,再次求解中间的值带入判断m^2与x之间的关系,一旦小于,我们用 i 将其记录下来,左边从中间加一,直到循环至左大于右结束。
举例:x=8:
第一次循环:l=0,r=8,m=4 4*4=16>8 r=m-1=3 此时:l=0,r=3,满足循环条件,进行第二次循环
第二次循环:l=0,r=3,m=1 1*1=1<8 i=m=1 l=m+1=2 满足第三次循环条件
第三次循环:l=2,r=3,m=2, 2*2=4<8 i=m=2 l=m+1=3 满足循环条件
第四次循环:l=3,r=3,m=3, 3*3=9>8 r=m-1=2 此时l>r跳出循环结束,返回 i 的值
代码:
class Solution(object):
def mySqrt(self, x):
l=0
r=x
while(l<=r):
m=(l+r)//2
if(m*m>x):
r=m-1
else:
i=m
l=m+1
return i
复杂度分析:
a.时间复杂度:O(logx)
b.空间复杂度:O(1)