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
要注意几个点:
- 终止条件
- 中值需要向下取整,可以使用移位操作。
left
和right
的变化- 最终返回的结果。
想想为什么返回值是 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=xn−f′(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)(x−x0), 则 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=x0−f′(x0)f(x0), 称 x 1 x_{1} x1 为 r r r 的一次近似值。
来源:牛顿迭代法
本次求平方根就相当于求解
f
(
x
)
=
x
2
−
a
f(x)=x^{2}-a
f(x)=x2−a 的零点,将方程带入上述公式,就可以得到求根的表达式:
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))