LeetCode Top Interview Questions 69. Sqrt(x) (Java版； Easy)

welcome to my blog

LeetCode Top Interview Questions 69. Sqrt(x) (Java版; Easy)

题目描述

Implement int sqrt(int x).

Compute and return the square root of x, where x is guaranteed to be a non-negative integer.

Since the return type is an integer, the decimal digits are truncated and only the integer part of the result is returned.

Example 1:

Input: 4
Output: 2
Example 2:

Input: 8
Output: 2
Explanation: The square root of 8 is 2.82842..., and since 
            the decimal part is truncated, 2 is returned.

class Solution {
    public int mySqrt(int x) {
        if(x<=1){
            return x;
        }
        //left, right指向大于平方根的最小值
        int left=1, right=x, mid;
        while(left<right){
            mid = left + (right - left)/2;
            if(mid > x / mid){
                right = mid;
            }else if(mid < x / mid){
                left = mid + 1;
            }else{
                return mid;
            }
        }
        return left-1;

    }
}

第一次做; 二分法; 最开始觉得不好用二分法, 因为没有考虑到二分到死

/*
核心: 
1. 证明 mid<x/mid 以及 mid>x/mid的有效性, 进而避免溢出
2. 二分循环条件, 会影响到返回值
3. 返回值的解释

细节:
1.不用追求right的初始值

证明mid<x/mid 以及 mid>x/mid的有效性
对于int类型的变量来说, x/mid的结果会舍弃小数部分, 也就是相当于x/mid的下取整, 用floor(x/mid)表示x/mid的下取整, 下面的证明过程都表示精确计算, 不会舍弃小数
(1) mid < floor(x/mid)时, 因为 floor(x/mid) <= x/mid, 所以推出mid < x/mid, 进一步得出mid*mid < x; 也就是说我们可以在程序中用mid < floor(x/mid)等效表示mid*mid < x, 这样做的好处是防止mid*mid溢出

(2) mid > floor(x/mid)时, 同时 floor(x/mid) <= x/mid, 看起来没法直接得到mid和x/mid的大小关系, 其实不然, 因为mid是整数, 所以mid能取到的最小值是floor(x/mid)+1, 与此时同, x/mid舍弃了小数部分得到floor(x/mid), 小数部分一定小于1, 所以x/mid的最大值一定小于floor(x/mid)+1, 进而得出mid > x/mid, 所以mid > floor(x/mid)能够等效表示mid*mid>x

(3) mid = floor(x/mid)时, 同时 floor(x/mid) <= x/mid, 说明mid <= x/mid, 进一步得出mid*mid <= x, 虽然不是mid*mid=x, 但是题目中舍弃了平方根的小数部分, 比如x=10, mid=3时, 满足mid*mid <= x, 此时floor(x/mid)=3=mid, 也是题目的解

证明过程不完备, 这道题不简单
*/
class Solution {
    public int mySqrt(int x) {
        //0需要单独处理, 否则mid可能会为0, 出现除以0的情况
        if(x==0)
            return 0;
        int left=1, right=x, mid;
        while(left<right){
            mid = left + ((right-left)>>1);
            //如果left, right, mid是long类型, 这里可以直接写成mid*mid<x, 不会有溢出问题
            if(mid < x/mid)
                left = mid + 1;
            else if(mid > x/mid)
                right = mid - 1;
            else
                return mid;
        }
        /*
        如果left*left>x, 就返回left-1;  如果left*left<=x, 就返回left
        解释:
        什么时候出现left==right?
        如果是right缩小1时出现了left==right, 说明原来的right平方一定大于x, 但是不知道当前的right平方是否满足条件, 所以判断一下再返回
        如果是left增加1时出现了left==right, 说明原来的left平方一定小于x, 当时不知道当前的left平方是否满足条件, 所以判断一下再返回
        */
        return left > x/left ? left - 1 : left; 
    }
}