审题
实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842...,
由于返回类型是整数,小数部分将被舍去。
这道题的看到之后感觉暴力轻松会AC吧,结果还是碰到了一点门槛,因为在结果是最大数的时候是没有办法进行 i ∗ i i*i i∗i操作的,当然,这个我们看下面的代码再进行说明了。
代码实现
方案一(上文提到的的方法):
这个方法就是很憨的了,暴力,次数是根号x吧。
class Solution {
public:
int mySqrt(int x) {
int res;
for ( int i=0; i<=x; i++ ) {
if ( i == 46341 ) {
res = 46340;
break;
}
if ( i*i < x ) continue;
else if ( i*i == x ) {
return i;
break;
} else {
res = i-1;
break;
}
}
return res;
}
};
方案2:
看了官方题解后发现二分查找原来是这么用的…原谅一个算法小白的无知。需要尤其注意的两个地方:
- mid*mid要用long long否则会出界
- 边界条件要看好,可以通过自己编的数据试一试(我现在是这么干的)
class Solution {
public:
int mySqrt(int x) {
int l=0, r=x, ans=-1, mid;
while ( l<=r ) {
mid = l + (r-l)/2;
if ( (long long)mid * mid <= x ) {
ans = mid;
l = mid + 1;
} else {
r = mid - 1;
}
}
return ans;
}
};
反思
二分查找对于时间复杂度的提升是大大滴,思想很好,这个简单题就算过了,明天我们搞一个中等题试一下。