实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842…,
由于返回类型是整数,小数部分将被舍去。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sqrtx
解题思路
一个数的平方肯定不会超过它的一半, 利用二分法来快速找到平方根
左边界初始化为 0, 右边将初始化为整型最大值
每次取出右中位数(左中位数会陷入死循环)
判断中位数的平方与 x 的大小关系
若大于 x, 则直接将右边界设置为中位数 - 1, 否则左边界设置为中位数
最后返回左边界或右边界(两个相等,因为循环条件是左边界小于右边界)
代码
class Solution {
public int mySqrt(int x) {
long left = 0;
long right = Integer.MAX_VALUE;
while (left < right) {
long mid = (left + right + 1) >>> 1;
long square = mid * mid;
if (square > x) {
right = mid - 1;
} else {
left = mid;
}
}
return (int) left;
}
}