69. x 的平方根

问题描述

• 给你一个非负整数 x ，计算并返回 x 的 算术平方根 。
• 由于返回类型是整数，结果只保留 整数部分 ，小数部分将被 舍去 。
• 注意：不允许使用任何内置指数函数和算符，例如 pow(x, 0.5) 或者 x ** 0.5 。

示例1：
输入：x = 4
输出：2

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

题目来源：LeetCode官方
链接地址：https://leetcode.cn/problems/sqrtx/description/

思路

• 题目很简单，就是让我们求平方根（这里不考虑负数，即算术平方根），我们的思路是从0出发，往数的正方向移动，找一个数n，使得它满足n * n<=x，且(n+1) * (n+1)>x，那么n就是我们要求的数，现在的问题来了，怎么去找这个n？
• 说到查找，我们可以想到几个查找算法，如我们可以按照上面所说，令i=0，分别判断i * i，(i+1) * (i+1)与x的关系，再i++即可，这就是顺序查找的思想,时间复杂度为O(n)。但这样效率未免有些低下，所以我们想到了二分查找，也称折半查找，时间复杂度为O(logn)，其具体思想在代码中体现
• 这里说明一下，我们的查找区间是[0,x]，因为0和1的平方都等于其本身，考虑到这种情况，所以两端取闭区间

核心代码

    /**
     * 使用二分查找法求根，注：只要求得到x的平方根的整数部分即可
     * @param x
     * @return
     */
    public static int mySqrt(int x) {
        int left = 0, right = x;
        int ans = 0;//x的平方根，先初始化为0
        while(left <= right){
            int mid = left + (right - left) / 2;//这里值与（left+right）/2相等，但这样写可以尽量避免数值溢出
            //我们要求的是x的平方根，那么就不断用某个数的平方去逼近x
            if ((long)mid * mid <= x){//如果mid的平方小于x，说明左边的值太小了，舍弃左边----这里加一个long，避免mid平方太大，超出int范围，导致出错
                ans = mid;//这里我们设定只有在 mid*mid<=x 时，才更新ans的值，因为ans是x平方根的整数部分，所以 ans*ans<x 必定成立
                left = mid + 1;//随着left的右移，会导致mid的增大，即ans的值会逐渐逼近x平方根的整数部分
            }else {
                right = mid - 1;
            }
        }
        return ans;
    }

    public static int MySqrt(int x) {
        int left = 0, right = x;
        int ans = 0;//x的平方根，先初始化为0
        while(left <= right){
            int mid = left + (right - left) / 2;//这里值与（left+right）/2相等，但这样写可以尽量避免数值溢出
            //我们要求的是x的平方根，那么就不断用某个数的平方去逼近x
            if ((long)mid * mid < x){//如果mid的平方小于x，说明左边的值太小了，舍弃左边
                left = mid + 1;//随着left的右移，会导致mid的增大，即ans的值会逐渐逼近x平方根的整数部分
            }else {
                ans = mid;//这里我们设定只有在 mid*mid>=x 时，才更新ans的值
                right = mid - 1;
            }
        }
        return ans-1;
    }

代码解释

这里有两个代码片段，二者都是采用二分查找，主要区别在于ans = mid这条语句的位置，导致了return语句的不同，大家可以自己弄个测试用例分别试一下就知道了为什么return会不同。假设n为要求的平方根，那第一个代码就是从0往n去逼近，第二个代码就是从x往n方向去逼近

运行效果

完整源代码

package com.easy;

public class mySqrt {
    public static void main(String[] args) {
        System.out.println(mySqrt(24));
        System.out.println(MySqrt(24));

    }

    /**
     * 使用二分查找法求根，注：只要求得到x的平方根的整数部分即可
     * @param x
     * @return
     */
    public static int mySqrt(int x) {
        int left = 0, right = x;
        int ans = 0;//x的平方根，先初始化为0
        while(left <= right){
            int mid = left + (right - left) / 2;//这里值与（left+right）/2相等，但这样写可以尽量避免数值溢出
            //我们要求的是x的平方根，那么就不断用某个数的平方去逼近x
            if ((long)mid * mid <= x){//如果mid的平方小于x，说明左边的值太小了，舍弃左边----这里加一个long，避免mid平方太大，超出int范围，导致出错
                ans = mid;//这里我们设定只有在 mid*mid<=x 时，才更新ans的值，因为ans是x平方根的整数部分，所以 ans*ans<x 必定成立
                left = mid + 1;//随着left的右移，会导致mid的增大，即ans的值会逐渐逼近x平方根的整数部分
            }else {
                right = mid - 1;
            }
        }
        return ans;//从左往右逼近，得到的是x平方根的向下取整值，刚好满足题目要求
    }

    public static int MySqrt(int x) {
        int left = 0, right = x;
        int ans = 0;//x的平方根，先初始化为0
        while(left <= right){
            int mid = left + (right - left) / 2;//这里值与（left+right）/2相等，但这样写可以尽量避免数值溢出
            //我们要求的是x的平方根，那么就不断用某个数的平方去逼近x
            if ((long)mid * mid < x){//如果mid的平方小于x，说明左边的值太小了，舍弃左边
                left = mid + 1;//随着left的右移，会导致mid的增大，即ans的值会逐渐逼近x平方根的整数部分
            }else {
                ans = mid;//这里我们设定只有在 mid*mid>=x 时，才更新ans的值
                right = mid - 1;
            }
        }
        return ans-1;//从右往左逼近，得到的是x平方根的向上取整值，会比题目要求的值大1，所以这里返回ans-1
    }
}

运行截图