[LeetCode] 69、x 的平方根

题目描述

实现 int sqrt(int x) 函数。计算并返回 x 的平方根，其中 x 是非负整数。由于返回类型是整数，结果只保留整数的部分，小数部分将被舍去。

输入: 8
输出: 2
说明: 8 的平方根是 2.82842..., 
     由于返回类型是整数，小数部分将被舍去。

解题思路

面试中遇到先说二分解法，这是程序员的思路，此外还有“牛顿法”。学AI的应该会。

• 二分查找：常规的二分，跳出来的时候一定是在平方根附近的，最后判断一下如果平方大于x的话就返回它前面的一个值，否则就正常返回就行了。（代码）

• 牛顿迭代法（题解1，题解2）：直接看题解中的“牛顿法”部分，这部分也需要会。

  • 简单介绍：下面这种方法可以很有效地求出根号 $a$的近似值：首先随便猜一个近似值 $x$，然后不断令 $x$等于 $x$ 和 $a/x$ 的平均数，迭代个六七次后 $x$ 的值就已经相当精确了。

    例如，我想求根号2 等于多少。假如我猜测的结果为 4，虽然错的离谱，但你可以看到使用牛顿迭代法后这个值很快就趋近于根号 2 了：

    ( 4 + 2/ 4 ) / 2 = 2.25

    ( 2.25 + 2/ 2.25 ) / 2 = 1.56944…

    ( 1.56944…+ 2/1.56944…) / 2 = 1.42189…

    ( 1.42189…+ 2/1.42189…) / 2 = 1.41423…

    这种算法的原理很简单，我们仅仅是不断用 $(x,f(x))$的切线来逼近方程$x^2-a=0$的根。

    

    根号$a$实际上就是 $x^2-a=0$的一个正实根，这个函数的导数是 $2x$。也就是说，函数上任一点 $(x,f(x))$处的切线斜率是 $2x$。那么，$x-\frac{f(x)}{2x}$ 就是一个比 $x$ 更接近根的近似值。代入 $f(x)=x^2-a$得到 $x-\frac{x^2-a}{2x}$，也就是 $\frac{x+a/x}{2}$。

参考代码

二分法（一）

class Solution {
public:
    int mySqrt(int x) {
        if(x == 0)
            return 0;
        if(x <= 3)
            return 1;
        
        int low = 1, high = x;
        while(low <= high){
            int mid = ((long long)low + high) >> 1;   // 加long long的地方全部都是溢出的问题
            long long temp = (long long)mid * mid;
            long long temp2 = (long long)(mid + 1) * (mid + 1);
            if(temp == x || (temp < x && temp2 > x))
                return mid;
            
            if(temp < x)
                low = mid + 1;
            else
                high = mid - 1;
        }
        
        return 0;
    }
    
};

二分法（二） 用模板写的，相对更简单。

typedef long long LL;
class Solution {
public:
	int mySqrt(int a) {
		if (a == 0 || a == 1)
			return a;

		int left = 1, right = a;
		while (left < right) {
			int mid = left + ((right - left + 1) >> 1);  // 有位运算符的地方，要注意加括号

			if ((LL)mid * mid > a)  // 注意这里也可能溢出
				right = mid - 1;
			else
				left = mid;
		}

		return left;
	}

};

牛顿法（要会）

class Solution {
public:
    int mySqrt(int a) {
        if(a == 0) 
            return 0;
        
        long long x = a;
        while(x * x > a)  // 找第一个小于等于 根号a 的数，所以初始的x可以设置的大一点
            x = (x + a / x) / 2;
        
        return (int)x;
    }
};