Python LeetCode 69. Sqrt(x) 二分法和牛顿迭代法
二分法
class Solution(object):
def mySqrt(self, x):
"""
:type x: int
:rtype: int
"""
l, r, result = 0, x, -1
while l <= r:
mid = (l + r) // 2
if mid * mid <= x:
result = mid
l = mid + 1
else:
r = mid - 1
return result
牛顿迭代法
https://leetcode-cn.com/problems/sqrtx/solution/niu-dun-die-dai-fa-by-loafer/
参考上面的链接,其实上面写的我看不太懂,之后弄懂了,就我写一下我自己的理解。借一下图
这个地方举例子非常好,很容易看出来算的很快
原文这个地方我是没太懂的一开始。
没懂的是x-f(x)/(2x)这一句,其实这一句,可以分开来看,f(x)/(2x)其实就是f(x)/f(x)的导数,那么几何含义就是变化量,
图片中红色的部分。之后就是代入公式化简了,注意先算/再算减号。
class Solution(object):
def mySqrt(self, x):
"""
:type x: int
:rtype: int
"""
if x == 0:
return 0
a, x0 = float(x), float(x)
while True:
xi = 0.5 * (x0 + a / x0)
if abs(x0 - xi) < 1e-7:
break
x0 = xi
return int(x0)