【算法】浮点数开方

题目

给一个浮点数$x>0$，计算$\sqrt{x}$，返回值也为float类型。

随想

之前一次面试，被问到这个题。当时忘了解法，用泰勒展开，折腾半天，啥也没搞出来，无奈写了二分法。这里写篇博客强记一下。

思路

牛顿法！牛顿法！牛顿法！
我直接把这个题泛化一下，假设是开$k$次方，$k\in \mathbb{N}$且$k>1$。为了防止搞混，记输入的$x$为$x_0$，那么实际上就是求下式的零点：
$$y=x^k-x_0\text{\hspace{1em}(1)}$$

注意构造的方式（面试的时候脑子木了好久），我们通过迭代的方式不断逼近真实结果，初始迭代直接采用$x_0$，记第$n$次迭代后为$x_n$，由图易得

$$x_{n+1}=x_n-\frac{y_n}{y_n^{\prime }}\text{\hspace{1em}(2)}$$

注意上式即为牛顿法的一般公式，牛顿法和泰勒展开的一阶近似求零点是一样的，即求下式的零点
$$y=f(x_n)+f(x_n^{\prime })(x-x_n)\text{\hspace{1em}(3)}$$

回到本题，把式(1)代入(2)，得到
$$x_{n+1}=(1-\frac{1}{k})x_n+\frac{x_0}{k}\frac{1}{x_n^{k-1}}=\frac{1}{k}[(k-1)x_n+\frac{x_0}{x_n^{k-1}}]\text{\hspace{1em}(4)}$$

当$k=2$时，上式为
$$x_{n+1}=\frac{1}{2}{x}_n+\frac{x_0}{2}\frac{1}{x_n}=\frac{1}{2}(x_n+\frac{x_0}{x_n})\text{\hspace{1em}(4)}$$

代码

float my_sqrt(float x, float epsilon) {  // x > 0; 正确结果和返回值之间的误差上界为epsilon
	float x0 = x;
	while (((x - epsilon) * (x - epsilon) > x0 && x > epsilon) 
			|| (x + epsilon) * (x + epsilon) < x0) {
			x = (x + x0 / x) / 2;
	}
	return x;
}