平方根函数

sqrt function

Posted by BY on March 23, 2018

前言

人蠢要多读书!更加要多动脑!!(To me.)

正文

问题来源

本问题来自leetcode 69题。刚看到这题的时候,就感觉这道题不难(不难的原因是想到直接暴力求解,^A^hehe)。做完这道题提交后,才发现效率不行,没有办法通过。于是乎,上网搜关于这道题的最优解法(优雅姿势)。

1.二分

思路:要实现一个sqrt函数,可以使用二分法,首先确定一个范围[begin, end],这个范围的中间数mid,看mid的平方是否等于x,如果相等,则返回mid,如果不等则缩小[begin,end]的范围,为原来的一半。这里的初始范围可以是[1, x],也可以是更精确一些的[1, (x/2) + 1]。(因 (x/2) + 1 的平方等于 x+1+(x^2/4),它一定大于x,所以,x的平方根一定在[1, (x/2) + 1]范围内)

题目中给出的函数原型是int mySqrt(int x)。参数和返回值都是整数。这里稍微扩展一下,将函数原型改为double mySqrt(int x)。解题思路还是一样的,但是浮点数因精度的原因,无法判断两个浮点数是否完全相等,只能说两者的差值绝对值小于某个精度,所以在二分查找时,需要一定的技巧,具体的代码如下:

double mySqrt_binarysearch(int x) 
{
	if(x <= 0)	return 0;

	double begin = 1;
	double end = x/2+1;
	double mid, lastmid;

	mid = begin + (end-begin)/2;
	do{
		if(mid < x/mid) begin = mid;
		else	end = mid;

		lastmid = mid;
		mid = begin + (end-begin)/2;
	}
	while(ABS(lastmid-mid) > FLT_MIN);
	
	return mid;
}

2.牛顿迭代法

上面的实现方法只能说是中规中矩,但是实现sqrt有更牛逼的方法,就是牛顿迭代法。该方法就是由我们熟知的牛顿提出的。具体思想可以自行搜索。简而言之,如下图:

x^2 = a的解,也就是函数f(x) = x^2 – a与x轴的交点。可以在x轴上先任选一点x0,则点(x0, f(x0))在f(x)上的切线,与x轴的交点为x1,它们满足切线的方程:f(x0)=(x0-x1)f’(x0),可得x1更接近最终的结果,解方程得到: x1 = (x0 + (a/x0))/2。以x1为新的x0,按照切线的方法依次迭代下去,最终求得符合精确度要求的结果值。它的实现代码如下:

double mySqrt_newton(int x) 
{
	if(x <= 0)	return 0;

	double res, lastres;

	res = x;	//初始值,可以为任意非0的值
	
	do{
		lastres = res;
		res = (res + x/res)/2;
	}
	while(ABS(lastres-res) > FLT_MIN);

	return res;
}

3.神迹

下面这段代码出自《雷神之锤》,至今尚未找到该代码的真正作者,代码如下:


float InvSqrt(float x)
{
    float xhalf = 0.5f * x;
    int i = *(int*)&x; 
    i = 0x5f375a86 - (i>>1); 
    x = *(float*)&i;
    x = x*(1.5f-xhalf*x*x); 
    x = x*(1.5f-xhalf*x*x); 
    x = x*(1.5f-xhalf*x*x);

    return 1/x;
}


它本质上还是使用的牛顿迭代法,真正牛逼的地方在于它初始值的选择,0x5f375a86这个魔法数字的由来尚不可知,该算法的具体原理及其背景可以参见维基百科,不再赘述。

要注意的是,上面算法使用的是float和int类型,实验可知他们不能替换为double和long。

结语

牛顿法-wiki

平方根倒数速算法-wiki

Sqrt函数引发的血案-博客园