题目描述
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例1
示例2
题解
思路:解ans是x平方根的整数部分,也就是求解k平方<=x的最大值,对x进行二分查找。
设解为ans,对x进行二分查找,对于二分值mid,如果mid平方<=x,那么ans可能是mid也可能比mid大,先记录ans=mid,再将二分头设置为mid+1,在mid之后继续查找ans;反之,操作类似。
重点在于如果(long long)mid * mid <= x,不但要将二分头重新设置为mid+1,还要记录ans=mid,因为很有可能这个mid值就是k**2<=x的最大k值。
还有就是取二分值mid时,如果用 mid = (l+r)/2 整数int的大小会溢出,因为 r 可能就是最大的 int 值,所以用mid = l + (r - l) / 2 表示,这是一样的。
代码
class Solution {
public:
int mySqrt(int x) {
int l = 0, r = x, ans = -1;
while (l <= r) {
int mid = l + (r - l) / 2;
if ((long long)mid * mid <= x) {
ans = mid;
l = mid + 1;
} else {
r = mid - 1;
}
}
return ans;
}
};