[Leetcode]x的平方根

[Leetcode]x的平方根

Leetcode-x的平方根

题目描述

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

示例 1:
输入: 4
输出: 2

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

解题思路

• 袖珍计算器算法
  • 「袖珍计算器算法」是一种用指数函数 $\exp$ 和对数函数 $\ln$ 代替平方根函数的方法。我们通过有限的可以使用的数学函数，得到我们想要计算的结果。
  • $y=\sqrt{x}=x^{1/2},\ln y=\ln x^{1/2},y=e^{1/2\ln x}$
  • 注意： 由于计算机无法存储浮点数的精确值（浮点数的存储方法可以参考 IEEE 754，这里不再赘述），而指数函数和对数函数的参数和返回值均为浮点数，因此运算过程中会存在误差。例如当 x = 2147395600时，$e^{\frac{1}{2}\ln x}$的计算结果与正确值 4634046340 相差 $10^{-11}$ ，这样在对结果取整数部分时，会得到 4633946339 这个错误的结果。因此在得到结果的整数部分 res 后，我们应当找出 res与 res+1 中哪一个是真正的答案。

class Solution:
    def mySqrt(self, x: int) -> int:
        if x == 0:
            return 0
        res = int(math.exp(0.5 * math.log(x)))
        return res + 1 if (res + 1) ** 2 <= x else res

• 二分查找
  • 由于 x 平方根的整数部分 ans 是满足 $k^2\le x$ 的最大 k 值，因此我们可以对 k 进行二分查找，从而得到答案。
  • 二分查找的下界为 0，上界可以粗略地设定为 x。在二分查找的每一步中，我们只需要比较中间元素 $\text{mid}$ 的平方与 x 的大小关系，并通过比较的结果调整上下界的范围。由于我们所有的运算都是整数运算，不会存在误差，因此在得到最终的答案 $\text{ans}$ 后，也就不需要再去尝试 $\text{ans}+1$ 了。

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