题目:
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4 输出: 2
示例 2:输入: 8 输出: 2
说明: 8 的平方根是 2.82842…,
由于返回类型是整数,小数部分将被舍去。
解题思路:
方法1(暴力解法):
(1)通过多次循环找出 x/n==n的值,或者是x/n <n的值,如果是后者,说明当前的元素2次方已经超出x值,需减去1;
(2)为了防止乘积溢出,用除法替代 n * n ==x的写法。
方法2(二分查找法)
代码实现:
class Solution:
def mySqrt(self, x: int) -> int:
if x == 0:
return 0
n = 1
while True:
if x / n == n: # 防止乘积溢出,用除法替代
return n
elif x / n < n:
return n - 1
else:
n = n + 1
def mySqrt2(self, x: int) -> int:
if x == 0 or x == 1:
return x
x1 = 1
x2 = x
while x1 <= x2:
mid = (x1 + x2) // 2
if x / mid == mid:
return mid
elif x / mid > mid: # 如果当前算出的数大于中间数,则说明需要在右侧元素区域进行选取
x1 = mid + 1
elif x / mid < mid: # 如果当前算出的数小于中间数,则说明需要在左侧元素区域进行选取
x2 = mid - 1
return x2