实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
输入: 8
输出: 2
说明: 8 的平方根是 2.82842…,
由于返回类型是整数,小数部分将被舍去。
leetCode链接:https://leetcode-cn.com/problems/sqrtx/
思路
1.明确目标 什么样的数字符合要求?
因为是平方根的整数部分,所以这个整数a的平方一定小于给定参数x,并且是符合要求的数字里面最大的一个数, 这样,我们就能定位到每一个给定参数x,只有一个正整数a符合要求
这里使用二分法算法快速定位到a
时间复杂度: O(logn)
空间复杂度: O(1)
代码如下:
class Solution {
public int mySqrt(int x) {
// 初始化参数 左右边界 和 正整数a
int l = 0, r = x, a = -1;
// 循环条件是 左边界要小于等于右边界
while (l <= r) {
// 计算得到中间值
int mid = l + r + 1 >>> 1;
// 如果中间值符合有要求,增左边界=中间值+1
// -》计算右半部分是否还有符合条件的,保证最终的 a 是符合条件的最大值
// 否则就计算左半边的数据是否有符合的(右半边已经全部不符合了)
if ((long) mid * mid <= x) {
a = mid;
l = mid + 1;
} else {
r = mid - 1;
}
}
// 返回最终结果
return a;
}
}
扩展
这道题目还有另一种解决方案 -》 牛顿迭代法
嗯… 这种小肖实在没看懂,后续补充吧,这是 题解链接