一、题目
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842...,
由于返回类型是整数,小数部分将被舍去。
二、解决
1、二分查找
思路:
思路相对清晰,就是不断二分,不过中间有好多细节需要注意,一不小心容易溢出。
然后这里附一下二分算法模板,如下:
// java
public int binarySearch(int[] array, int target) {
int left = 0, right = array.length - 1, mid;
while (left <= right) {
mid = (right - left) / 2 + left;
if (array[mid] == target) {
return mid;
} else if (array[mid] > target) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return -1;
}
代码-版本1:
class Solution {
public int mySqrt(int x) {
if (x == 0 || x == 1) return x;
// Binary Search
int left = 0, right = x;
while (left < right) {
// mid = (left + right) / 2 can overflow if right > Integer.MAX_VALUE - left
int mid = left + (right - left) / 2;
// same thing here , mid * mid > x can overflow. replace by mid > x / mid
if (mid > x / mid) {
right = mid - 1;
} else {
left = mid + 1;
// if mid * mid < x but (mid + 1) * (mid + 1) > x then mid was the right answer
if (left > x / left) {
return mid;
}
}
}
return left;
}
}
代码-版本2:
class Solution {
public int mySqrt(int x) {
long left=0, right=x, mid;
while (left <= right) {
mid = left + (right-left)/2;
if ( mid*mid == x ) {
return (int)mid;
} else if (mid*mid > x) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return (int)right;
}
}
时间复杂度: O ( l o g n ) O(logn) O(logn)
空间复杂度: O ( 1 ) O(1) O(1)
2、牛顿迭代
思路:
由上图可知,过 ( x 0 , f ( x 0 ) ) (x_0, f(x_0)) (x0,f(x