[leetcode] 69. x 的平方根
文章目录
- 题目描述
- 解题方法
- 方法一:二分查找
- java代码
- 复杂度分析
- 方法二:牛顿迭代法
- java代码
- 复杂度分析
题目描述
给你一个非负整数 x
,计算并返回 x
的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5)
或者 x ** 0.5
。
示例 1:
输入:x = 4
输出:2
示例 2:
输入:x = 8
输出:2
解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
提示:
- 0 <= x <= 231 - 1
解题方法
方法一:二分查找
题目说了x
在[0, 231-1]之间,那么我们取左边界l
为0,右边界r
为x
,取中间值mid
计算乘积是否小于等于x
。若小于等于则l = mid + 1
,否则r = mid -1
;一直取mid
计算乘积不断遍历,直到左边界超过右边界,此时右边界r
即为最接近x
的平方根。
java代码
public int mySqrt(int x) {int l = 0, r = x;while (l <= r) {int mid = l + (r - l)/ 2;// mid相乘可能超过int整形范围,故值取long类型做比较long result = (long)mid * (long)mid;if (result > (long) x) {r = mid - 1;} else {l = mid + 1;}}return r;
}
复杂度分析
时间复杂度: O ( l o g x ) O(logx) O(logx)
空间复杂度: O ( 1 ) O(1) O(1)
方法二:牛顿迭代法
我们设 f ( x ) = x 2 − C f(x) = x^2 - C f(x)=x2−C, C C C为题目中输入,求函数的零点即为 C C C的平方根。因为 x x x是非负整数,所以 x x x的平方根一定小于等于 x x x,在单根附近具有平方收敛的特性,可以使用牛顿迭代法求解。
牛顿迭代法的公式如下:
x n + 1 = x n − f ( x n ) f ′ ( x n ) x_{n+1} = x_n - \frac{f(x_n)}{f'(x_n)} xn+1=xn−f′(xn)f(xn)
我们将 f ( x ) = x 2 − C f(x) = x^2 - C f(x)=x2−C带入上述方程,即可求出
x n + 1 = 1 2 ∗ ( x n + C x n ) x_{n+1} = \frac{1}{2} * (x_n + \frac{C}{x_n}) xn+1=21∗(xn+xnC)
借用下力扣题解图,我们取 x 0 = C x_0 = C x0=C,带入公式即可求出 x 0 x_0 x0点的切线与 x x x轴的交点 x 1 x_1 x1,带入 x 1 x_1 x1求出 x 2 x_2 x2,依次类推,直到 x n + 1 x_{n+1} xn+1与 x n x_{n} xn非常接近时,我们即取 x n + 1 x_{n+1} xn+1作为最终解。
java代码
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;
}
复杂度分析
时间复杂度: O ( l o g x ) O(logx) O(logx)
空间复杂度: O ( 1 ) O(1) O(1)
- 个人公众号
- 个人小游戏