题目描述
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842…,
由于返回类型是整数,小数部分将被舍去。
思路
解法1
遍历1到x/2的数,注意判断是不要用 “i*i == x” 这种,因为会溢出。
解法2
在解法1的基础上调整为二分查找法,降低时间复杂度
解法3
百度搜索得到的方法,牛顿迭代法
代码
解法1
class Solution {
/**
* 遍历1到x/2的数,注意判断是不要用 “i*i == x” 这种,因为会溢出。
*/
public int mySqrt(int x) {
if(x ==1 || x == 0){
return x;
}
int halfX = x / 2;
int ret = 0;
for(int i = 1; i <= halfX; ++ i){
if(i <= x / i && (i+1) > x / (i + 1)){
ret = i;
break;
}
}
return ret;
}
}
解法2
class Solution {
/**
* 在解法1的基础上调整为二分查找法,降低时间复杂度
*/
public int mySqrt(int x) {
if(x <= 1){
return x;
}
int ret = 0;
int start = 0;
int end = x;
while(start < end){
int mid = start + (end - start) / 2;
if(mid <= x / mid)
start = mid + 1;
else
end = mid;
}
return end - 1;
}
}
解法3
class Solution {
/**
* 牛顿迭代法,同样避免 ret * ret = x的判断条件
*/
public int mySqrt(int x) {
if(x == 0){
return x;
}
long ret = (long)x;
while(ret > x / ret){
ret = (ret + x/ret) / 2;
}
return (int)ret;
}
}