思路:二分查找算法
参考题解,这里mid的取法值得注意。
public static int mySqrt(int x) {
int left = 0;
int right = x / 2 + 1 ; // 否则会超时
// 在区间 [left..right] 查找目标元素
while (left < right) {
int mid = left + (right - left) / 2; // mid取为上界,即mid <= right, 而不是mid >= left
// 否则会出现mid == left, 然后一直循环下去
// 注意:这里为了避免乘法溢出,改用除法
if (mid > x / mid) {
// 下一轮搜索区间是 [left..mid - 1]
right = mid;
} else {
// 下一轮搜索区间是 [mid..right]
left = mid + 1;
}
}
return left;
}