机器学习面试必知:牛顿法实现sqrt

牛顿法是一种常用的求方程数值解,具体方法如下
若在区间 I I I中, f ( x ) f(x) f(x)连续可导,且有唯一零点 x 0 x_{0} x0,则任取 x 1 ∈ I x_{1}\in I x1I,定义数列 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=xnf(xn)f(xn)经过多次迭代后 x n x_{n} xn会趋向于 x 0 x_{0} x0

输入是 x x x,求 t = x → t 2 − x = 0 t=\sqrt{x}\rightarrow t^{2}-x=0 t=x t2x=0,那么假设有函数 f ( t ) = t 2 − x f(t)=t^{2}-x f(t)=t2x,求函数零点。 t n + 1 = t n − f ( t n ) f ′ ( t n ) = t n 2 + x 2 t n t_{n+1}=t_{n}-\frac{f(t_{n})}{f^{'}(t_{n})}=\frac{t_{n}}{2}+\frac{x}{2t_{n}} tn+1=tnf(tn)f(tn)=2tn+2tnx

class Solution:
    def mySqrt(self, x: int) -> int:
        t=x
        while abs(t*t-x)>=1e-6:
            t=t/2+x/(2*t)
        return t

你可能感兴趣的:(机器学习,面试)