python——Leetcode 69. x 的平方根

题目：

实现 int sqrt(int x) 函数。

计算并返回 x 的平方根，其中 x 是非负整数。

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

示例 1:

输入: 4
输出: 2
示例 2:

输入: 8
输出: 2
说明: 8 的平方根是 2.82842..., 
     由于返回类型是整数，小数部分将被舍去。

思路是二分法
我的最初答案：

class Solution:
    def mySqrt(self, x):
        """
        :type x: int
        :rtype: int
        """
        if x==0:
            return 0
        if x==1:
            return 1
        min = 0
        max = x
        mid = (min+max)//2
        while max-min>1:
            m = mid*mid
            if m>x:
                max = mid
                mid = (min+max)//2
            elif m<x:
                min = mid
                mid = (min+max)//2
            else:
                return mid
                break
        return min

我的这个运行时间比较长，原因可能是最开始判断了0和1的情况，而且没有很恰当的使用二分法。通过学习别人的代码，发现自己的不足，代码如下：

class Solution:
    def mySqrt(self, x):
        """
        :type x: int
        :rtype: int
        """
        min = 0
        max = x
        mid = x//2
        while min<=max:
            m = mid*mid
            if m>x:
                max = mid-1
            elif m<x:
                min = mid+1
            else:
                return mid
                break
            mid = (min+max)//2
        return mid