x 的平方根:简单问题不简单

leetcode 69:x 的平方根

返回的是只保留整数的部分

二分查找

该题是一个很经典的二分查找算法。

def mySqrt(self, x: int) -> int:
    left, right = 1, x
    while left <= right:
        mid = left + right >> 1   # 取整,相当于 // 2,除以二取整。
        temp = mid ** 2
        if temp == x:
            return mid
        elif temp > x:
            right = mid - 1
        else:
            left = mid + 1
    return right

要注意几个点:

  • 终止条件
  • 中值需要向下取整,可以使用移位操作。
  • leftright 的变化
  • 最终返回的结果。

想想为什么返回值是 right 而不是 left?
不用 left 这一点,找几个数演算一下就行了,它取得往往是向上取整,至于为什么我也不太懂。
希望有懂的人指教一下。

扩展: 求 x 的平方根的准确值
def mySqrt(self, x: int) -> int:
    left, right = 0, x
    while right - left > 1e-6:  # 计算精度,即退出条件
        mid = (left + right) / 2  # 除法
        temp = mid ** 2
        if temp == x:
            return mid
        elif temp > x:
            right = mid  # 区间要连续变化
        else:
            left = mid
  
    return right
牛顿法

牛顿迭代公式:
x n + 1 = x n − f ( x n ) f ′ ( x n ) x_{n+1}=x_{n}-\frac{f\left(x_{n}\right)}{f^{\prime}\left(x_{n}\right)} xn+1=xnf(xn)f(xn)
就是通过多次求经过某一点的切线与横坐标的交点而得到的。

r r r f ( x ) = 0 f(x)=0 f(x)=0 的根, 选取 x 0 x_{0} x0 作为 r r r 的初始近似值, 过点 ( x 0 , f ( x 0 ) ) \left(x_{0}, f\left(x_{0}\right)\right) (x0,f(x0)) 做曲线 y = f ( x ) y=f(x) y=f(x) 的切线 L L L L : y = f ( x 0 ) + f ′ ( x 0 ) ( x − x 0 ) , L: y=f\left(x_{0}\right)+f^{\prime}\left(x_{0}\right)\left(x-x_{0}\right), L:y=f(x0)+f(x0)(xx0), L L L x x x 轴交点的横坐标 x 1 = x 0 − f ( x 0 ) f ′ ( x 0 ) , x_{1}=x_{0}-\frac{f\left(x_{0}\right)}{f^{\prime}\left(x_{0}\right)}, x1=x0f(x0)f(x0), x 1 x_{1} x1 r r r 的一次近似值。
来源:牛顿迭代法

本次求平方根就相当于求解 f ( x ) = x 2 − a f(x)=x^{2}-a f(x)=x2a 的零点,将方程带入上述公式,就可以得到求根的表达式:
x n + 1 = x n + a / x n 2 x_{n+1}=\frac{x_{n}+a / x_{n}}{2} xn+1=2xn+a/xn

代码如下:

# 牛顿法
def mySqrt(self, x: int) -> int:
    r = x
    while r ** 2 - x >= 1:  # 设置精度,r 最终退出时,其平方与准确值之间小于 1。然后取整数部分即可。
        r = (r + x/r) / 2
    return int(r)
内置函数(偷懒法)
from math import sqrt
def mySqrt(self, x: int) -> int:
    return int(sqrt(x))
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值