原题链接:leetcode-cn.com/problems/sq…
解题思路:
- 要理解牛顿迭代法,我们需要先回顾一下导数,或者可以看这一篇导数入门。
- 题解参考了二分查找 + 牛顿法(Python 代码、Java 代码)、牛顿迭代法、69. x 的平方根-二分查找, 牛顿法。
- 根据题意,该题是这样一个方程,已知求。可以将其用函数代替。
- 这个函数的导数是,也就是。
- 将带入,可以得到。
- 将带入,可以得到。
- 将等式整理,可以得到。
/**
* @param {number} x
* @return {number}
*/
var mySqrt = function (x) {
function sqrt(x0) {
// 当遇到第一个x0 * x0 <= x的值时,表示已经得到最接近的结果,将x0的整数部分返回即可
// 用Math.floor将x0*x0向下取整
// 避免例如TestCase: 5,会出现x0 * x0=5.000000000000001,造成死循环
if (Math.floor(x0 * x0) <= x) {
return Math.floor(x0);
}
// 套用分析得到的迭代公式,计算出新的x0
x0 = (x0 + x / x0) / 2;
return sqrt(x0);
}
return sqrt(x);
};