leetcode x 的平方根 python

 x 的平方根

 

 

实现 int sqrt(int x) 函数。

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

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

示例 1:

输入: 4
输出: 2

示例 2:

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

这道题如果用python的开方函数那就没有意思了。所以我手动实现了一下。
我觉着这道题的思想是一个二分查找。

 class Solution(object):
     def mySqrt(self, x):
         """
         :type x: int
         :rtype: int
         """
         if x > 1:
             return self.search(0, x, x)
         elif x == 1 or x == 0:
             return x
         else:
             pass

     def search(self, start, end, x):
         mid = int((start + end) / 2)
         mid2 = mid * mid
         mid22 = (mid + 1) * (mid + 1)
         if mid2 <= x < mid22:
             return mid
         elif mid2 > x:
             return self.search(start, mid, x)
         else:
             return self.search(mid, end, x)