给你一个非负整数 x
,计算并返回 x
的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5)
或者 x ** 0.5
。
示例 1:
输入:x = 4 输出:2
示例 2:
输入:x = 8 输出:2 解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
提示:
0 <= x <= 231 - 1
解法:二分查找法
class Solution {
public int mySqrt(int x) {
if (x == 0) {
return 0;
}
if (x == 1){
return 1;
}
int left = 1;
int right = x / 2;
// 在left和right区间查找目标元素
while (left < right) {
int mid = left + (right - left + 1) / 2;
// 注意:这里为了避免乘法溢出,改用除法
if (mid > x / mid) {
// 下一轮搜索区间是 [left..mid - 1]
right = mid - 1;
} else {
// 下一轮搜索区间是 [mid..right]
left = mid;
}
}
return left;
}
}
官方解答:
方法一:袖珍计算器算法
「袖珍计算器算法」是一种用指数函数 exp和对数函数 ln代替平方根函数的方法。我们通过有限的可以使用的数学函数,得到我们想要计算的结果。
我们将 根号x 写成幂的形式 x的1/2次方,再使用自然对数 e 进行换底,即可得到
这样我们就可以得到 根号x 的值了。
注意: 由于计算机无法存储浮点数的精确值(浮点数的存储方法可以参考 IEEE 754,这里不再赘述),而指数函数和对数函数的参数和返回值均为浮点数,因此运算过程中会存在误差。例如当 x=2147395600时,e^1/2lnx 的计算结果与正确值 46340 相差 10^−11 ,这样在对结果取整数部分时,会得到 46339 这个错误的结果。
因此在得到结果的整数部分 ans后,我们应当找出 ans与 ans+1 中哪一个是真正的答案。
class Solution {
public int mySqrt(int x) {
if (x == 0) {
return 0;
}
int ans = (int) Math.exp(0.5 * Math.log(x));
return (long) (ans + 1) * (ans + 1) <= x ? ans + 1 : ans;
}
}
复杂度分析
时间复杂度:
O(1),由于内置的 exp 函数与 log 函数一般都很快,我们在这里将其复杂度视为 O(1)。
空间复杂度:
O(1)。
作者:力扣官方题解
链接:https://leetcode.cn/problems/sqrtx/