需求
给你一个非负整数 x ,计算并返回 x 的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5) 或者 x ** 0.5 。
(leetcode.69)
我的思路
基于二分法。
判断当前索引和下一个索引的平方值与target的大小。如,若target = 10,3 * 3 < 10而4 * 4 > 10,则输出3。
注意:int类型的平方可能越界,需要采用long
public int mySqrt(int x) {
int r = x / 2;
int max = x;
int result = -1;
if(x == 0 || x == 1) {
return x;
}
while(r <= x) {
if(mul(r) > x) {//太大了,减半
if(mul(r - 1) < x) {
return r - 1;
}
max = r;
r = r / 2;
} else if(mul(r) < x) {//太小了,增大
if(mul(r + 1) > x) {
return r;
}
r = (r + max) / 2;
} else {
return r;
}
}
return -1;
}
public long mul(int i) {
return (long) i * i;
}
心得:对于计算 i*i ,相比于直接运算,将其封装为方法会消耗更少的时间。(我当时是直接运算显示超时,封装后才通过)
答案思路
public int mySqrt(int x) {
int l = 0;
int r = x;
while(l < r) {
int mid = (l + r) / 2;
if((long)mid * mid < x) {
l = mid + 1;
} else {
r = mid;
}
}
return (long)r*r <= x ? r : r - 1;
}