实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4 输出: 2
示例 2:
输入: 8 输出: 2 说明: 8 的平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
方法:
二分查找
二分查找的下界为 0,上界可以设定为 x。在二分查找的每一步中,我们只需要比较中间元素 mid 的平方与 x 的大小关系,并通过比较的结果调整上下界的范围。
代码:
//69. x 的平方根
static void test_69() {
//int x = 4;
//int x = 8;
//int x = 1;
int x = 2147395599;
int ret = mySqrt(x);
System.out.println("ret:" + ret);
}
static int mySqrt(int x) {
int l = 0;
int r = x;
int ans = 0;
while (l <= r) {
int mid = l + (r - l)/2;
// 输入为2147395599,mid*mid溢出,long(mid*mid)防止溢出
if ((long)mid*mid <= x) {
ans = mid;
l = mid + 1;
} else {
r = mid - 1;
}
}
return ans;
}