【Leetcode 刷题记录】- 69. x 的平方根（20220711 ）

凯迪拉克麦旋风

已于 2022-07-11 11:28:03 修改

阅读量399

点赞数

分类专栏：力扣算法题堆积处文章标签：算法

于 2022-07-11 11:27:28 首次发布

本文链接：https://blog.csdn.net/m0_46578941/article/details/125716355

版权

力扣算法题堆积处专栏收录该内容

4 篇文章 1 订阅

订阅专栏

求x的平方根有多种解法，以下是解题记录和总结。

法1. 二分法

法2. 牛顿迭代法

法3. 不动点法求平方根

题目：

给你一个非负整数 x ，计算并返回 x 的算术平方根。

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

注意：不允许使用任何内置指数函数和算符，例如 pow(x, 0.5) 或者 x ** 0.5 。

示例 1：

输入：x = 4
输出：2
示例 2：

输入：x = 8
输出：2
解释：8 的算术平方根是 2.82842..., 由于返回类型是整数，小数部分将被舍去。

提示：

0 <= x <= 231 - 1

法1. 二分法

主要思想：（零点存在定理）通过比较零点和中值的位置关系，来判断边界的移动。

第一步：定义函数f(x)：

private:
    int fx(long long int num, int diff){
        return diff * diff - num;
    }

第二步：判断函数f(x)的零点关系。

 int mySqrt(long long int x) {
        //使用二分法
        if(x == 0){
            return 0;
        }
        else if(x == 1){
            return 1;
        }

        int left = 1;
        int right = x - 1;
        while(left < right){
            int middle = left + ((right - left + 1) / 2);
            if ( fx(x, middle) == 0){
                return middle;
            }
            else if( fx(x, middle) < 0 ){
                left = middle;
            }
            else{
                right = middle - 1;
            }
        }
        return left;
    }

但是在使用上述方法求解时，会报一个溢出错误：

Line 31: Char 21: runtime error: signed integer overflow: 1073697800 * 1073697800 cannot be represented in type 'int' (solution.cpp) SUMMARY: UndefinedBehaviorSanitizer: undefined-behavior prog_joined.cpp:40:21

因此做出如下优化：

class Solution {
public:
    int mySqrt(long long int x) {
        //使用二分法
        if(x == 0){
            return 0;
        }
        else if(x == 1){
            return 1;
        }

        int left = 1;
        int right = x - 1;
        while(left < right){
            int middle = left + ((right - left + 1) / 2);
            if (middle == x / middle){
                return middle;
            }
            else if(middle < x / middle){
                left = middle;
            }
            else{
                right = middle - 1;
            }
        }
        return left;
    }

};

法2. 牛顿迭代法

算法介绍：

设 $r$ 是 $f(x)=0$ 的根，选取 $x_{0}$ 作为 $r$ 的初始近似值，则我们可以过点 $\left(x_{0}, f\left(x_{0}\right)\right)$ 做曲线 $y=f(x)$ 的切线 $L$ ,我们知道切线 $L$ 与 $x$ 轴有交点，我们已知切线 $L$ 的方程为 $L: y=f\left(x_{0}\right)+f^{\prime}\left(x_{0}\right)\left(x-x_{0}\right)$ 我们求的它与 $x$ 轴的交点为 $x_{1}=x_{0}-\frac{f\left(x_{0}\right)}{f^{\prime}\left(x_{0}\right)}$ . 我们在以 $\left(x_{1}, f\left(x_{1}\right)\right)$ 斜率为做斜线 $f^{\prime}\left(x_{1}\right)$ ，求出与 $x$ 轴的交点，重复以上过程直到 $f\left(x_{n}\right)$ 无限接近于0即可。其中第n次的迭代公式为：

$x_{n+1}=x_{n}-\frac{f\left(x_{n}\right)}{f^{\prime}\left(x_{n}\right)}$

题目分析：本题中可设 $f(d)=d \times d - x$ ，因此有 $d_{n+1}=d_{n}-\frac{f\left(d_{n}\right)}{f^{\prime}\left(d_{n}\right)}$ 。

计算得到： $d_{n+1}=\frac{d_{n}^{2} + x}{2d_{n}}$

代码实现：

class Solution {
public:
    int mySqrt(long long int x) {
        if (x==0) return 0;
        long diff = x;
        while (diff * diff > x) diff = (diff + x / diff) / 2;
        return (int)diff;
    }
};