69. x的平方根

题目

刚开始写的：

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

结果爆int了。

改成这样：

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

就通过了。

附上分析过程：

不过官方的二分题解更好理解：

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;
    }
};

相信看了这个代码应该能体会到其中的妙处，感觉像是分成了两个区间和一个点，并且把这个点单独考虑。