69. x 的平方根
实现 int sqrt(int x)
函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
思路:有了之前对二分模板的理解,不难画出下图
要注意的几点:
- 尽管题目给出的
x
是int
类型,但是我们计算mid
时需要l+r+1
,故如果x
是Integer.MAX_VALUE
则会出现越界,故我们定义l与r的时候选择long
型。 mid * mid <= x
中,mid * mid
也和1中一样,可能越界,故写成mid <= x / mid
,至于x / mid
会向下取整的问题,题目给出的结果也是要去除小数部分的,故无影响。- 最终的结果需要转回
int
Java代码
class Solution {public int mySqrt(int x) {long l = 0,r = x;//既然是求平方根,结果肯定在0~x里面while(l < r){long mid = (l + r + 1) / 2;if(mid <= x / mid) l = mid;else r = mid - 1;}return (int)l;}
}