leetcode69. x 的平方根

184 阅读1分钟

69. x 的平方根

实现 int sqrt(int x) 函数。计算并返回 x 的平方根,其中 x 是非负整数。

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

示例 :

输入: 8
输出: 2
说明: 8 的平方根是 2.82842..., 
     由于返回类型是整数,小数部分将被舍去。
class Solution:
    def mySqrt(self, x: int) -> int:
        lo = 0  # 二分法,因为y=x**2图像为单调递增,有序
        hi = x
        while lo <= hi:
            mid = (hi + lo) // 2
            y = mid * mid
            if y < x:
                lo = mid + 1
            elif y > x:
                hi = mid - 1
            else:
                return mid
        return hi  # 注意结束条件,结束时hi<lo,取整数hi为答案