给你一个非负整数 x ,计算并返回 x 的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5) 或者 x ** 0.5 。
示例 1:
输入:x = 4
输出:2
示例 2:
输入:x = 8
输出:2
解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
提示:
0 <= x <= 231 - 1
来源:leetcodeT69
思路:
首先这个数开方的结果肯定小于自身,并且小于等于自身的一半,所以有了上边界和下边界,可以使用二分查找,除0和1、2外,仅需事先进行判断即可
核心:目标值n^2<=x<(n+1)^2
所以二分查找的方法中,可分为主要三类:
a、mid^2 == x 可直接返回结果为mid
b、mid^2 > x 可明确得到结果为mid不是目标值,并且大了,需要将上边界改为mid
c、mid^2 < x 有两种可能,
第一种,mid不是目标值,需要将 下边界调整到mid处
第二种,mid 是目标值,因为mid+1的平方已经大于x,不能再增大了
所以可以先通过条件:(mid+1)^2 > x 做二次判断,如果成立,那么直接返回mid为结果
如果不成立,则将下边界调整为mid继续循环
代码如下:
public static int test(int x) {
if (x == 0) {
return 0;
} else if (x == 1) {
return 1;
}
int lo = 1;
int hi = x / 2;
while (lo < hi) {
int mid = (lo + hi) / 2;
if (x / mid == mid) {
return mid;
} else if (x / mid < mid) {//mid*mid > x,一定不是,需要缩小
hi = mid;
} else {//mid*mid <x,可能是可能不是,
//如何判断是?(mid+1)^2>x,那么x就是
if (x / (mid + 1) < (mid + 1)) {
//是mid
return mid;
}else if(x / (mid + 1) == (mid + 1)){
return mid+1;
} else {
lo = mid;
}
}
}
return 1;//当x为2的时候,会出现一开始就上下边界重合的情况,所以返回1
}
进行判断mid 和 x的关系时,使用的是x/mid的方式,可避免mid*mid数值溢出的问题