本关的编程任务是补全右侧代码片段中 Begin 至 End 中间的代码，编写一个能求出整数 x 平方根的小程序，具体要求如下： 使用二分查找，实现 int mySqrt(int x) 函数，求平方根的功能。计算并返回 x 的平方根，其中 x 是非负整数。 注：由于返回类型是整数，结果只保留整数的部分，小数部分将被舍去。 测试举例 测试输入： 4 预期输出： 2

class Solution:
    def mySqrt(self, x: int) -> int:
        if x == 0 or x == 1:
            return x
        left, right = 1, x
        while left <= right:
            mid = (left + right) // 2
            if mid * mid == x:
                return mid
            elif mid * mid < x:
                left = mid + 1
                ans = mid
            else:
                right = mid - 1
        return ans   # Begin