实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例
输入: 4
输出: 2
class Solution:
def mySqrt(self, x: int) -> int:
l, r, ans = 0, x, -1
while l <= r:
mid = (l + r) // 2
if mid * mid <= x:
ans = mid
l = mid + 1
else:
r = mid - 1
return ans
由于本题不允许使用sqrt等一系列函数,所以用最传统的方法,从零开始不断找出一个数用它的平方与x比较知道找到最接近的整数值。
所以要找到这个值,可以用到查找的方法—二分查找。
定义一个上界和一个下界,上界为x下界为0,并定义一个变量mid用以存储上界与下界之间的中间值,当且仅当下界不在小于上界时结束,返回结果。
这里一开始没有想到二分查找法,直接采取下界到上界的循环遍历,会造成时间和空间的浪费