Leetcode sqrt 平方根

sqrt(x)

Implement int sqrt(int x).

Compute and return the square root of x.

x is guaranteed to be a non-negative integer.

'''
Solution: 二分法，判断要求：mid²和x的关系
'''

def sqrt(x):
    if x <= 0:
        return -1
    
    left, right = 1, x
    while left + 1 < right:
        mid = left + (right - left) // 2       
        if mid**2 == x:
            return mid
        elif mid**2 < x:
            left = mid  #此处二分法使用的环境不同于重复数组中找特定数字，因此使用+1，-1可以缩短时间
        else:
            right = mid
    
    #此处循环条件left + 1 < right表明了循环结束时left + 1 == right
    #对某个数开根取整只是向下取整，因此选小的那个数即可，也就是right
    #取小的那个，即left
    return left

sqrt(121)

11

version 2:

def sqrt(x):
    if x == 0:
        return 0
    left, right = 1, x
    while left <= right:
        mid = left + (right - left) // 2
        
        print(left, right, mid)
        
        if mid == x // mid: #这么写比mid**2 == x好的一点就是可以防止mid**2过大导致溢出
            return mid
        elif mid < x // mid:
            #易错点：这里一定要mid ± 1， 如果只是mid的话，可能会导致新一轮的mid和上一层的mid一样，从而导致死循环
            left = mid + 1
        else:
            right = mid - 1
        
    #这里要取right是因为判断条件left <= right表明了循环结束时left>=right，
    #而对某个数开根取整只是向下取整，因此选小的那个数即可，也就是right
    return right