x 的平方根

x 的平方根

题目:
实现 int sqrt(int x) 函数。

计算并返回 x 的平方根，其中 x 是非负整数。

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

示例 1:

输入: 4
输出: 2
示例 2:

输入: 8
输出: 2
说明: 8 的平方根是 2.82842…,
由于返回类型是整数，小数部分将被舍去。

解题思路: 二分即可, 需要注意的是l, r, mid需要定义为long

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