69. Sqrt(x)

问题

Implement int sqrt(int x).

Compute and return the square root of x.

例子

sqrt(8)
2

分析

注意到sqrt的返回值为int，不是float或者double，这样问题就简单很多了。而且经过测试可以发现，s = sqrt(x)，s应该是满足s*s<=x的最大值。

使用二分法可以在O(logn)的时间找到一个整数的整数平方根。

要点

二分法（lower_bound）

时间复杂度

O(logn)

空间复杂度

O(1)

代码

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

牛顿迭代法

class Solution {
public:
    int mySqrt(int x) {
        long long r = x;
        while (r*r > x)
            r = (r + x/r) / 2;
        return r;
    }
};