实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:输入: 8
输出: 2
说明: 8 的平方根是 2.82842…, 由于返回类型是整数,小数部分将被舍去。
思路:二分查找
- 设定查找的边界, 根据查找范围中间值的平方相较于 x 的大小, 将边界不断收缩到原来的 1 / 2
- 定义 left = 1,right = Math.floor(x/2)两个指针为结果的左右边界。因为结果不会超过x/2,所以right指向x的一半向下取整,缩小遍历范围 。mid = Math.floor((left + right) / 2) .
- 若 mid * mid > x 说明平方根在 mid 左侧,则更新right和mid指针,right = mid; mid = Math.floor((left +right) / 2);
- 若 mid * mid < x 说明平方根在 mid 右侧,则更新left和mid指针,left = mid;mid = Math.floor((left + right) / 2);
- 若mid * mid == x 输出mid;
- 循环结束的条件是当left和right相邻时,再判断一次左右指针选择输出结果
/**
* @param {number} x
* @return {number}
*/
var mySqrt = function(x) {
if(x<2) return x;
let left = 1;
let right = x >> 1;
let mid;
while(left + 1 < right){//left+1=right时,左右相邻,退出循环
mid = (left+right)>>1;
if(mid > x/mid){//用mid > x/mid代替mid*mid>x防止溢出
right = mid;
}else if(mid < x/mid){
left = mid;
}else{
return mid;
}
}
return right > x/right ? left:right;
};
复杂度分析:
时间复杂度: O(logn), n = x. 查找过程中, 查找范围不断减半.
空间复杂度: O(1)