题目
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842...,
由于返回类型是整数,小数部分将被舍去。
思路
二分搜索 [0, x] 区间,找到从左往右第一个平方大于等于 x 的数 ans,最后判断如果 ans 的平方等于 x,就输出 ans,如果 ans 的平方大于 x,输出 ans - 1。
时间复杂度:
O
(
log
x
)
O(\log x)
O(logx)
空间复杂度:
O
(
1
)
O(1)
O(1)
C++ 代码
class Solution {
public:
int mySqrt(int x) {
long long left = 0, right = x, mid;
while (left < right) {
mid = left + (right - left) / 2;
if (mid * mid >= x)
right = mid;
else
left = mid + 1;
}
return left * left == x ? left : left - 1;
}
};