题目描述
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
不能够for循环遍历
//i * i 会溢出int类型
int i;
for(i = 0;i < x;i++){
if(i * i > x){
break;
}
}
return i;
二分法
//既然不能够用int类型,我们可以使用long类型,时间复杂度降为O(logx)
class Solution {
public int mySqrt(int x) {
int l = 0, r = x, ans = -1;
while (l <= r) {
int mid = l + (r - l) / 2;
if ((long) mid * mid <= x) {
ans = mid;
l = mid + 1;
} else {
r = mid - 1;
}
}
return ans;
}
}
牛顿法逼近
牛顿迭代法的本质是借助泰勒级数,从初始值开始快速向零点逼近。我们任取一个 x0 作为初始值,在每一步的迭代中,我们找到函数图像上的点 (xi, f(xi)),过该点作一条斜率为该点导数 f′(xi)的直线,与横轴的交点记为xi+1。xi+1相较于 xi而言距离零点更近。在经过多次迭代后,我们就可以得到一个距离零点非常接近的交点。下图给出了从 x0开始迭代两次,得到 x1和 x2的过程。
那么目的就是计算x1,x2的值。从而渐渐逼近目标值,这就是1级泰勒级数的应用。
我们应该选择选择 x0=C 作为初始值
具体看下面题解
https://leetcode-cn.com/problems/sqrtx/solution/x-de-ping-fang-gen-by-leetcode-solution/
class Solution {
public int mySqrt(int x) {
if (x == 0) {
return 0;
}
double C = x, x0 = x;
while (true) {
double xi = 0.5 * (x0 + C / x0);
if (Math.abs(x0 - xi) < 1e-7) {
break;
}
x0 = xi;
}
return (int) x0;
}
}