LeetCode69

LeetCode初体验

感觉很“友善”，题目很简单，但是我是真的菜，长路漫漫。

题目

实现 int sqrt(int x) 函数。
计算并返回 x 的平方根，其中 x 是非负整数。
由于返回类型是整数，结果只保留整数的部分，小数部分将被舍去。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sqrtx
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码

/**
 * Solution69
 */
public class Solution69 {

    public static void main(String[] args) {
        Solution slt = new Solution();
        int w = slt.mySqrt(2147483647);
        System.out.println(w);
    }
}

/**
 * Solution
 */
class Solution {
    public int mySqrt(int x) {
        // 负数报异常
        // if (x < 0)
        //     throw new IllegalArgumentException("Illegal Capacity: " + x);

        // 小于等于1直接求
        if (x <= 1) {
            return x;
        }

        // 二分法，使用Long防止int溢出
        long n = 1, m = x, mid;
        do {
            mid = (m+n)/ 2;
            if(mid*mid>x){
                m = mid;
            }
            else{
                n = mid;
            }
        }while(m-n>1);
        return (int)n;
    }
}

关键点

1. 使用二分法时间代价应该是最小的了；
2. 一开始n、m、mid3个中间变量用的是int类型,但是用加法、乘法，数过大时会溢出，只好改为long，最终结果强制转换回int;
3. 后来，在室友提示下，可以用减法改加法算式，用除法改乘法算式避免溢出！mid = (m+n)/2改为mid = m-(m-n)/2,mid*mid>x改为mid>x/mid,这样每步计算不会超过最大的int数,但是除法运算慢，也算是内存空间消耗和时间代价的trade off;
4. 最后可以把除以2改成循环右移一位，速度更快。