leetcode 69. x 的平方根

可以使用二分查找法或牛顿迭代法来实现 LeetCode 问题 69. x 的平方根。下面是使用二分查找法和牛顿迭代法的 C++ 实现。

二分查找法

cpp 复制代码
#include <iostream>

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

int main() {
    Solution solution;
    int x = 8;
    std::cout << "The square root of " << x << " is " << solution.mySqrt(x) << std::endl;
    return 0;
}

牛顿迭代法

cpp 复制代码
#include <iostream>

class Solution {
public:
    int mySqrt(int x) {
        if (x == 0) return 0;
        double x0 = x;
        while (true) {
            double xi = 0.5 * (x0 + x / x0);
            if (abs(x0 - xi) < 1e-7) break;
            x0 = xi;
        }
        return static_cast<int>(x0);
    }
};

int main() {
    Solution solution;
    int x = 8;
    std::cout << "The square root of " << x << " is " << solution.mySqrt(x) << std::endl;
    return 0;
}

解释

二分查找法
  1. 初始化 :定义 left 为 1,rightx,并初始化 ans 为 0。
  2. 循环 :当 left 小于等于 right 时,计算 mid 作为中间值。
  3. 判断 :如果 mid 的平方小于等于 x,说明 mid 可能是平方根的一部分,更新 ansmid,并移动 leftmid + 1。否则,移动 rightmid - 1
  4. 返回 :循环结束后,返回 ans
牛顿迭代法
  1. 初始化 :定义 x0x
  2. 迭代 :计算 xi,它是 x0x / x0 的平均值。如果 x0xi 的差异小于一个很小的值(如 1e-7),则停止迭代。
  3. 更新 :将 x0 更新为 xi
  4. 返回 :将 x0 转换为整数并返回。

这两种方法都能有效地计算 x 的平方根,并且二分查找法的时间复杂度为 O(log x),牛顿迭代法的时间复杂度为 O(log x)。你可以根据需要选择其中一种方法。

当然,使用图示和例子可以更直观地理解二分查找算法在计算平方根整数部分的过程。

例子:计算 10 的平方根的整数部分

我们以计算 10 的平方根为例,来展示整个过程。

步骤 1:初始化
  • left = 1
  • right = 10
  • ans = 0
步骤 2:开始二分查找
  1. 第一次迭代

    • 计算中点 mid = left + (right - left) / 2 = 1 + (10 - 1) / 2 = 5

    • 检查 mid * midx 的关系:5 * 5 = 25,25 > 10,因此更新 rightmid - 1,即 right = 4

    • 图示:

      复制代码
      搜索区间: [1, 10]
      mid = 5, 5*5 > 10, 更新right = 4
  2. 第二次迭代

    • 计算中点 mid = left + (right - left) / 2 = 1 + (4 - 1) / 2 = 2

    • 检查 mid * midx 的关系:2 * 2 = 4,4 < 10,因此更新 ansmid,并更新 leftmid + 1,即 left = 3

    • 图示:

      复制代码
      搜索区间: [1, 4]
      mid = 2, 2*2 < 10, 更新left = 3, ans = 2
  3. 第三次迭代

    • 计算中点 mid = left + (right - left) / 2 = 3 + (4 - 3) / 2 = 3

    • 检查 mid * midx 的关系:3 * 3 = 9,9 < 10,因此更新 ansmid,并更新 leftmid + 1,即 left = 4

    • 图示:

      复制代码
      搜索区间: [3, 4]
      mid = 3, 3*3 < 10, 更新left = 4, ans = 3
  4. 第四次迭代

    • 计算中点 mid = left + (right - left) / 2 = 4 + (4 - 4) / 2 = 4

    • 检查 mid * midx 的关系:4 * 4 = 16,16 > 10,因此更新 rightmid - 1,即 right = 3

    • 图示:

      复制代码
      搜索区间: [4, 4]
      mid = 4, 4*4 > 10, 更新right = 3
结束循环

left > right 时,退出循环,此时 ans 保存的就是最大的满足条件的整数。最终结果为 ans = 3,所以 10 的平方根的整数部分是 3。

代码对应的流程

  1. 初始化 leftrightans
  2. 在每次迭代中计算 mid 并比较 mid * midx
    • 如果 mid * mid 小于等于 x,则更新 ans 并右移 left
    • 如果 mid * mid 大于 x,则左移 right
  3. 循环结束后,返回 ans

图示

复制代码
初始区间: [1, 10]

第一次迭代:
mid = 5, 5*5 > 10, 更新right = 4
搜索区间变为: [1, 4]

第二次迭代:
mid = 2, 2*2 < 10, 更新left = 3, ans = 2
搜索区间变为: [3, 4]

第三次迭代:
mid = 3, 3*3 < 10, 更新left = 4, ans = 3
搜索区间变为: [4, 4]

第四次迭代:
mid = 4, 4*4 > 10, 更新right = 3
搜索区间变为: [4, 3]

循环结束,返回 ans = 3

这样,通过二分查找,我们成功找到并返回了 10 的平方根的整数部分 3。

相关推荐
凯子坚持 c12 分钟前
深度解析算法之滑动窗口
数据结构·算法
安然无虞22 分钟前
31天Python入门——第17天:初识面向对象
后端·爬虫·python·职场和发展
程序员小远35 分钟前
Python+requests实现接口自动化测试框架
自动化测试·软件测试·python·测试工具·职场和发展·测试用例·接口测试
xyliiiiiL40 分钟前
二分算法到红蓝染色
java·数据结构·算法
珹洺2 小时前
C++从入门到实战(十)类和对象(最终部分)static成员,内部类,匿名对象与对象拷贝时的编译器优化详解
java·数据结构·c++·redis·后端·算法·链表
写bug的小屁孩2 小时前
移动零+复写零+快乐数+盛最多水的容器+有效三角形的个数
c++·算法·双指针
飞川撸码2 小时前
【LeetCode 热题100】208:实现 Trie (前缀树)(详细解析)(Go语言版)
算法·leetcode·golang·图搜索算法
这就是编程3 小时前
自回归模型的新浪潮?GPT-4o图像生成技术解析与未来展望
人工智能·算法·机器学习·数据挖掘·回归
羑悻的小杀马特3 小时前
【狂热算法篇】探寻图论幽径:Bellman - Ford 算法的浪漫征程(通俗易懂版)
c++·算法·图论·bellman_ford算法
Fantasydg6 小时前
DAY 31 leetcode 142--链表.环形链表
算法·leetcode·链表