题目:
给你一个非负整数 x ,计算并返回 x 的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5) 或者 x ** 0.5 。
示例 1:
输入:x = 4
输出:2
示例 2:
输入:x = 8
输出:2
解释:8 的算术平方根是 2.82842…, 由于返回类型是整数,小数部分将被舍去。
提示:
0 <= x <= 231 - 1
思路:
由于 x 平方根的整数部分 ans 是满足 k 2 ≤ x k^2 ≤x k2≤x 的最大 k 值,因此我们可以对 k 进行二分查找,从而得到答案。
二分查找的下界为 0,上界可以粗略地设定为 x。在二分查找的每一步中,我们只需要比较中间元素 middle 的平方与 x 的大小关系,并通过比较的结果调整上下界的范围。由于我们所有的运算都是整数运算,不会存在误差,因此在得到最终的答案 ans 后,也就不需要再去尝试 ans+1 了。
代码:
class Solution {
public:
int mySqrt(int x) {
int left = 0, right = x, ans = 0;
while(left <= right)
{
int middle = (left + right) / 2;
// x平方根的整数部分ans是满足k^2 ≤ x 的最大k值,因此我们可以对k进行二分查找,从而得到答案。
// 在C++中,对两个整数执行乘法运算时,结果会根据操作数的数据类型进行计算。
// 如果两个操作数都是int类型,结果也会是int类型。这意味着如果乘法的结果超出了int类型的范围,就会发生整数溢出。
//(long long)middle * middle的目的是确保乘法操作的结果不会因为溢出而丢失信息。
if((long long)middle * middle <= x)
{
ans = middle;
left = middle + 1;
}
else right = middle - 1;
}
return ans;
}
};
总结:
时间复杂度:O(logx),即为二分查找需要的次数。
空间复杂度:O(1)。
比较中间元素 middle 的平方与 x 的大小关系,并通过比较的结果调整上下界的范围。