【三次过】Lintcode 141. x的平方根

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/majichen95/article/details/82720718

实现 int sqrt(int x) 函数,计算并返回 x 的平方根。

样例

sqrt(3) = 1

sqrt(4) = 2

sqrt(5) = 2

sqrt(10) = 3

挑战

O(log(x))


解题思路:

由于x平方根一定是小于等于x/2,所以可以看作在0-x/2的有序数组中寻找出一个满足条件的数,这个条件是:

mid*mid<=x && (mid+1)*(mid+1)>x

自然的就想到二分搜索,但是注意mid*mid这样的操作很容易溢出,所以这里改变一下形式:

mid<=x/mid && (mid+1)>x/(mid+1)
public class Solution {
    /**
     * @param x: An integer
     * @return: The sqrt of x
     */
    public int sqrt(int x) {
        // write your code here
        if(x == 0)
            return 0;
        int l = 1;
        int r = 1 + x/2;
        
        while(l < r){
            int mid = (r-l)/2+l;
            
            if(mid<=x/mid && (mid+1)>x/(mid+1))
                return mid;
            else if(mid > x/mid){
                r = mid - 1;
            }else{
                l = mid + 1;
            }
        }
        
        return l;
    }
}

OR

public class Solution {
    /**
     * @param x: An integer
     * @return: The sqrt of x
     */
    public int sqrt(int x) {
        // write your code here
        if(x == 0)
            return 0;
        int l = 1;
        int r = 1 + x/2;
        int mid = 1;
        
        while(l <= r){
            mid = (r-l)/2+l;
            
            if(mid<=x/mid && (mid+1)>x/(mid+1))
                return mid;
            else if(mid > x/mid){
                r = mid - 1;
            }else{
                l = mid + 1;
            }
        }
        
        return mid;
    }
}

猜你喜欢

转载自blog.csdn.net/majichen95/article/details/82720718
今日推荐