LeetCode二分查找x 的平方根

题目描述

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

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

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

示例 1:

输入x = 4
输出2

示例 2:

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

解题思路

经典写法

二分查找的思想,通过不断缩小搜索范围,最终找到平方根整数部分。

为什么采用long而不是int

因为再计算r可能出现溢出

需要注意得一点是:在计算 mid使用 left+(right-left) / 2,而不是 (left + right) / 2。这样可以避免溢出

代码

class Solution {
    public int mySqrt(int x) {
        long high = x / 2;
        long low = 0;

        if (x == 1) {
            return 1;
        }

        while (low <= high) {
            long mid = low + (high - low) / 2;
            long r = mid * mid;

            if (r > x) {
                high = mid - 1;
            } else if (r < x) {
                low = mid + 1;
            } else {
                return (int) mid;
            }
        }

        // 返回 low - 1 而不是 low
        return (int) (low - 1);
    }
}

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注