题目描述:
给你一个非负整数 x ,计算并返回 x 的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5) 或者 x ** 0.5 。
示例:
示例 1:
输入:x = 4
输出:2
示例 2:
输入:x = 8
输出:2
解释:8 的算术平方根是 2.82842…, 由于返回类型是整数,小数部分将被舍去。
解答:
class Solution {
public int mySqrt(int x) {
if (x == 0) return 0;
int left = 1, right = x;
while (left <= right) {
int mid = left + (right - left) / 2;
if (mid <= x / mid && (mid + 1) > x / (mid + 1)) { // 找到了答案
return mid;
} else if (mid > x / mid) { // 继续在左侧查找
right = mid - 1;
} else { // 继续在右侧查找
left = mid + 1;
}
}
return left; // 如果没有找到答案,返回 left - 1
}
}
笔记:
首先剔除和1和0时的情况,return x
使用折半二分法查找,左left为1,右right为x,进行while循环
在left<=right的时候,寻找中间节点 int mid = left + (right - left) / 2;
注:
在二分查找中,如果使用 (left + right) / 2 来计算中间节点 mid,当 left 和 right 的值都很大时,left
+right 可能会导致整数溢出,从而得到错误的结果。
为了避免这个问题,我们可以使用 left + (right - left) / 2 来计算 mid,这样可以保证 mid 的值一定在 left 和 right 的中间,避免了整数溢出的问题。
例如(199+200)/2=299,超出199~200之间,所以用199+(200-199)/2=199,在范围内,有效控制溢出。
然后计算平方是否小于x
当mid的平方<=x,mid+1的平方>=x时,mid为题目要求的算术平方根的答案
但是
注:
因为在计算 mid * mid 时,如果 mid 的值过大,可能会导致整数溢出的问题,即结果超出了 int 类型的范围,从而得到错误的结果。
为了避免这个问题,我们可以把 mid * mid 和 x 相比较,如果 mid * mid 小于等于 x,则说明 mid 可能是平方根,否则mid 太大了,需要在左侧继续查找。因此,我们使用 mid <= x / mid 来判断 mid 是否为平方根,这样可以保证 mid * mid 不会超出 int类型的范围,避免了整数溢出的问题。