问题描述
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842…,
由于返回类型是整数,小数部分将被舍去。
链接:https://leetcode-cn.com/problems/sqrtx
问题分析
首先考虑到一个数的平方根的整数部分通常情况下不会大于她的一半,使用二分法比较合适,当然1是个特例,所以后续要考虑到1
解法一(二分法)
class Solution:
def mySqrt(self, x: int) -> int:
if x == 0: # 0就直接拉出来输出了
return 0
l = 1 # 考虑到1的特殊性,边界从1开始,如果开始就是1就直接不进循环
r = x // 2 # r为右边界
while l < r:
# 注意:这里一定取右中位数,如果取左中位数,代码可能会进入死循环
mid = l + (r - l + 1) // 2 # l即左边界,括号中计算的是目前左边界和右边界中的中
square = mid * mid # 如果中值的平方比原数小,则将左边界变成中值
if square > x:
r = mid - 1
else:
l = mid
return l
执行用时 :44 ms, 在所有 Python3 提交中击败了98.08%的用户
内存消耗 :13.1 MB, 在所有 Python3 提交中击败了95.42%的用户
时间复杂度:
O
(
l
o
g
N
)
O(logN)
O(logN),二分法的时间复杂度是对数级别的。
空间复杂度:
O
(
1
)
O(1)
O(1),使用了常数个数的辅助空间用于存储和比较
解法二(牛顿法)
牛顿法更详细的可以查看知乎马同学的回答:如何通俗易懂地讲解牛顿迭代法求开方?数值分析? - 马同学的回答 - 知乎
class Solution:
def mySqrt(self, x: int) -> int:
if x==0:
return x
xn=1 # 随机设置的初始值
while True:
temp=xn
xn=(xn+x/xn)/2
if abs(xn-temp)<1e-6:
return int(xn)
执行用时 :48 ms, 在所有 Python3 提交中击败了95.32%的用户
内存消耗 :13 MB, 在所有 Python3 提交中击败了98.38%的用户
第一次的错误写法:
当时虽然想都了二分,但是仅仅二分了一次,直接在循环里遍历一半数据,这个不可取,如果数比较大,运行时间太长
class Solution:
def mySqrt(self, x: int) -> int:
mid=x//2
i=0
while i<=mid+1: # 直接从0开始遍历,这个在数比较大的情况下太费时间,不可取
if i*i>x:
return i-1
break
elif i*i==x:
return i
break
i += 1