题目描述
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842...,
由于返回类型是整数,小数部分将被舍去。
题目分析
这类问题存在一个隐藏的有序条件,因为自然数本身就是从小到大的有序排列,所以可以使用「二分查找」进行问题的处理。
根据经验,一个自然数的「平方根」,不会超过其值的一半,所以就确定了搜索区间的右边界int right = x / 2
。在搜索过程中,如果当前值的平方大于目标值,则当前值一定不是我们需要的结果,需要将$right$
值来移动到当前值的左侧,即right = mid - 1
。
参考代码
class Solution {
public int mySqrt(int x) {
if (x == 0) {
return 0;
}
if (x == 1) {
return 1;
}
// 一个数的平方根最多不会超过它的一半
int left = 0;
int right = x / 2;
while (left < right) {
int mid = left + (right - left + 1) / 2;
if (Math.pow(mid, 2) > x) {
right = mid - 1;
} else {
left = mid;
}
}
return left;
}
}
ps:
- 可以看到在我们移动左右指针时,出现了
$left = mid$
的等式,则计算中间位置的$mid$
就一定是int mid = left + (right - left + 1) / 2
的计算方法。
复杂度分析:
- 时间复杂度:O(logN),因为使用了二分法进行数据规模的缩小。
- 空间复杂度:O(1),借助了常数个辅助空间进行存储和比较。