给你一个非负整数 x
,计算并返回 x
的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5)
或者 x ** 0.5
。
示例 1:
输入:x = 4 输出:2
示例 2:
输入:x = 8 输出:2 解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
提示:
0 <= x <= 231 - 1
解题思路
本题规定了不允许使用任何内置指数函数和算符,所以不能调函数取整。
那只能使用暴力遍历的方式,但不要傻傻的从x一个一个往回检查,可以二分法来进行遍历。这样可以减少时间复杂度。但不能直接判断吧mid * mid是否等于x,这样会溢出,所以考虑通过对x和mid的商的比较,得到答案,即判断x/mid 的值是否为平方根。考虑到平方根可能为一个小数,题目要求我们取整,所以如果x_sqrt > mid && x_sqrt_next < mid + 1,这个mid就是平方根取整后的数
解题代码
class Solution {
public:
int mySqrt(int x) {
if(x == 0 || x == 1) return x;
int left = 0,right = x;
while(left <= right){
int mid = (left + right) / 2;
int x_sqrt = x / mid;
int x_sqrt_next = x / (mid + 1);
if(x / x_sqrt == x_sqrt) return x_sqrt;
if(x_sqrt > mid && x_sqrt_next < mid + 1) return mid;
if(x_sqrt > mid) left = mid + 1;
if(x_sqrt < mid) right = mid - 1;
}
return 0;
}
};