给你一个非负整数 x ,计算并返回 x 的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5) 或者 x ** 0.5 。
示例 1:
输入:x = 4
输出:2
示例 2:
输入:x = 8
输出:2
解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
提示:
0 <= x <= 231 - 1
法一、投机
class Solution(object):
def mySqrt(self, x):
"""
:type x: int
:rtype: int
"""
return int(pow(x,0.5))
法二、运行速度极慢,超1000ms
class Solution(object):
def mySqrt(self, x):
"""
:type x: int
:rtype: int
"""
i=1
while i*i<=x:
i+=1
return i-1
法三、二分法,关键处还是有点生疏
def mySqrt(self, x):
l = 0 # 定义左指针为0
r = x + 1 # 定义右指针为x+1,加1是为了避免在二分查找中漏掉右边界x本身
while l < r: # 当左右指针未相撞时执行循环
mid = (l + r) // 2 # 计算当前指针中点的值
if mid ** 2 == x: # 如果当前中点的平方等于x,则直接返回该中点
return mid
if mid ** 2 < x: # 如果当前中点的平方小于x,说明整数平方根应该在mid的右侧,将左指针l更新到mid+1
l = mid + 1
else: # 否则整数平方根应该在mid的左侧,将右指针r更新到mid
r = mid
return l - 1 # 直到左右指针相撞后,返回左指针减1,即为最接近但小于x的整数平方根