LeetCode 69.x的平方根
题目描述
给你一个非负整数 x
,计算并返回 x
的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5)
或者 x ** 0.5
。
示例 1:
输入:x = 4
输出:2
示例 2:
输入:x = 8
输出:2
解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
提示:
0 <= x <= 2^31 - 1
思路
由于非负整数的平方属于有序数组且各不相同,本题可以使用二分查找,在1与x这个区间内进行查找。由于结果不可能是x本身,所以查找区间可以是[1,x),笔者选择使用左闭右开式二分查找的写法。
也可以观察规律,当某个数i的平方恰好大于x时,那么这个数减1就是x算术平方根的整数部分,例如8的算术平方根是2√2,而2<2√2<3,2√2的整数部分正好是2,但是这种方法效率较低。
代码
C++版:
方法一:
class Solution {
public:
int mySqrt(int x) {
for(int i=1;;i++){
// 使用x/i<i能够防止i*i溢出
if(x/i<i){
return i-1;
}
}
}
};
方法二:
class Solution {
public:
int mySqrt(int x) {
//剪枝
if(x<=1){
return x;
}
// 所找结果一定在0与x之间,可使用二分查找
int left=1;
int right=x;
while(left<right){
int middle=left+(right-left)/2;
if(x/middle<middle){
right=middle;
}
else if(x/middle>middle){
left=middle+1;
}
else{
return middle;
}
}
return left-1;
}
};
Python版:
方法一:
class Solution:
def mySqrt(self, x: int) -> int:
if x<=1 :
return x
i=1
while True :
if x//i<i:
return i-1
i+=1
方法二:
class Solution:
def mySqrt(self, x: int) -> int:
if x<=1 :
return x
left=1
right=x
while left<right:
middle = left+(right-left)//2
if x//middle<middle:
right=middle
elif x//middle>middle:
left=middle+1
else:
return middle
return left-1
需要注意的地方
1.注意不能直接比较middle*middle<x,因为int*int可能会溢出。