69. x 的平方根 二分查找 x为整数和浮点数的情况

实现 int sqrt(int x) 函数。
计算并返回 x 的平方根，其中 x 是非负整数。
由于返回类型是整数，结果只保留整数的部分，小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842..., 
     由于返回类型是整数，小数部分将被舍去。

解题思路

需要向上取整

class Solution:
    def mySqrt(self, x: int) -> int:
        if x == 0:
            return 0

        left = 1
        right = x // 2

        while left < right:
            # 注意：这里一定取右中位数，如果取左中位数，代码可能会进入死循环
            # mid = left + (right - left + 1) // 2
            mid = (left + right + 1) >> 1
            square = mid * mid

            if square > x:
                right = mid - 1
            else:
                left = mid
        # 因为一定存在，因此无需后处理
        return left

当x为浮点数时，注意x在0 ~ 1之间的情况，这个时候结果不在0~x之间。

import sys 
# x is float
# threshold 为边界阈值
def mySqrt(x, threshold):
    if x>1:
        left, right = 0, x 
    else:
        # 当x小于1时，要注意边界条件，因为这时0~x之间任何数的平方都小于x，不可能达到条件
        left, right = 0, 1
    mid = (left + right) / 2 
    while  abs(mid*mid-x)>threshold:
        if mid*mid>x:
            right = mid 
        else:
            left = mid 
        mid = (left + right) / 2
    return mid 
        
if __name__ == "__main__":
    print(mySqrt(4.0, 0.001))