69Sqrt(x)

69.Sqrt(x)

给你一个非负整数 x ，计算并返回 x 的 算术平方根 。

由于返回类型是整数，结果只保留整数部分 ，小数部分将被舍去 。

注意：不允许使用任何内置指数函数和算符，例如 pow(x, 0.5) 或者 x ** 0.5 。

示例1

输入：x = 4
输出：2

示例2

输入：x = 8
输出：2
解释：8 的算术平方根是 2.82842..., 由于返回类型是整数，小数部分将被舍去。

提示

$$0<=x<=2^{31}-1$$

代码

class Solution {
	public:
	    int mySqrt(int x) {
	        if(x == 0 || x == 1)
	        {
	            return x;
	        }
	
	        int left = 0;
	        int right = x;
	
	        while (left <= right)
	        {
	            int mid = left + (right - left) / 2;
	
	            // 可防止溢出
	            if(x / mid > mid)       //mid * mid < x
	            {
	                left = mid + 1;
	            }
	            else if(x / mid < mid)  //mid * mid > x
	            {
	                right = mid - 1;
	            }
	            else                    //mid * mid = x
	            {
	                return mid;
	            }
	        }
	        return right;
	    }     
	};

思考

1.用x/mid与mid进行比较，而不是mid*mid与x比较，这样可以防止溢出。
当然，对mid*mid进行数据类型转换，也是完全OK的

2.跳出循环后，为什么返回的是right?
本题其实是寻找满足$$a^2<x$$的最大整数a，当跳出循环时，left > right，而right其实就是满足条件的那个值。

改进

欣赏一下简洁的官方代码！

class Solution {
	public:
	    int mySqrt(int x) {
	        int l = 0, r = x, ans = -1;
	        while (l <= r) {
	            int mid = l + (r - l) / 2;
	            if ((long long)mid * mid <= x) {
	                ans = mid;
	                l = mid + 1;
	            } else {
	                r = mid - 1;
	            }
	        }
	        return ans;
	    }
	};

复杂度分析

时间复杂度：O(logx)，即为二分查找需要的次数。

空间复杂度：O(1)。