实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去
链接:https://leetcode-cn.com/problems/sqrtx
搜索一个解,首先想到的是二分法,后来看别人答案发现还可以用牛顿法,牛顿法可以用来求方程的根、求最优化问题的解,有必要复习一下哇。
class Solution:
def mySqrt(self, x: int) -> int:
mydict = {0:0,1:1,2:1,3:1,4:2,5:2,6:2,7:2,8:2,9:3,10:3,11:3,12:3,13:3,14:3,15:3,16:4}
if x in mydict:
return mydict[x]
lower = 4
upper = x//4
while lower <= upper:
mid = (lower+upper)//2
if mid**2>x:
upper = mid-1
elif (mid+1)**2<=x:
lower = mid+1
else:
return mid