[LeetCode]69. x 的平方根

题目

实现 int sqrt(int x) 函数。

计算并返回 x 的平方根，其中 x 是非负整数。

由于返回类型是整数，结果只保留整数的部分，小数部分将被舍去。

示例 1:

输入: 4
输出: 2

示例 2:

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

解题思路

解法一：二分法

由于 x 平方根的整数部分 res 是满足 k^2≤x 的最大 k 值，因此我们可以对 k 进行二分查找，从而得到答案。二分查找的下界为 0，上界可以粗略地设定为 x。在二分查找的每一步中，我们只需要比较中间元素 mid 的平方与 x 的大小关系，并通过比较的结果调整上下界的范围。

复杂度分析：
时间复杂度：O(logx)，即为二分查找需要的次数。
空间复杂度：O(1)。

解法二：牛顿迭代法：

我们用 C 表示待求出平方根的那个整数。显然，C 的平方根就是函数y = f(x) = x^2 - C 的零点。在每一步迭代中，我们通过当前的交点 x_i，找到函数图像上的点 (x_i, x_i^2 - C)，作一条斜率为 f'(x_i) = 2*x_i的直线，直线的方程为：y = 2x_i*(x - x_i) + x_i^2 - C = 2x_i*x - (x_i^2 + C)，与横轴的交点为方程 2x_i*x - (x_i^2 + C) = 0 的解，即为新的迭代结果x_{i+1}：x_{i+1} = (1/2)*(x_i + C/x_i)，在进行 k 次迭代后，x_k的值与真实的零点足够接近，即可作为答案。

复杂度分析：
时间复杂度：O(logx)，此方法是二次收敛的，相较于二分查找更快。
空间复杂度：O(1)。

代码

解法一：二分法

Python代码如下：

class Solution:
    def mySqrt(self, x: int) -> int:
        left = 0
        right = x
        res = 0
        while left<=right:
            mid = left + (right-left)//2
            if mid*mid<=x:
                res = mid
                left = mid+1
            else:
                right = mid-1
        return res

Java代码如下：

class Solution {
    public int mySqrt(int x) {
        int left = 0;
        int right = x;
        int res = 0;
        while(left<=right){
            int mid = left + (right-left)/2;
            if((long)mid*mid<=x){
                res = mid;
                left = mid+1;
            }else{
                right = mid-1;
            }
        }
        return res;
    }
}

解法二：牛顿迭代法

Python代码如下：

class Solution:
    def mySqrt(self, x: int) -> int:
        if x==0:
            return 0
        pre = x
        while True:
            cur = 0.5*(pre+x/pre)
            if abs(pre-cur)<1e-7:
                break
            pre = cur
        return int(pre)

Java代码如下：

class Solution {
    public int mySqrt(int x) {
        if(x==0){
            return 0;
        }
        double pre = x;
        while(true){
            double cur = 0.5*(pre+x/pre);
            if(Math.abs(pre-cur)<1e-7){
                break;
            }
            pre = cur;
        }
        return (int)pre;
    }
}