90. 实现 int sqrt(int x) 函数

实现 int sqrt(int x) 函数。

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

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

示例 1:

输入: 4
输出: 2

示例 2:

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

“””

如果不调用系统的sqrt函数,那有一个方法来实现,就是从1开始计算,直到计算平方大于x时,上一个数字就是 平方根。 代码如下:

import unittest

class Solution:
    def mySqrt(self, x):
        """
        :type x: int
        :rtype: int
        """
        tmp = 1
        for i in range(1, x//2+1):
           if(i*i<=x):
               tmp = i
           else:
               break;
        return tmp

class TestmySqrtFunc(unittest.TestCase):
    """Test case"""

    def test_mySqrt(self):
        """Test method mySqrt(self, x)"""
        s = Solution()
        self.assertEqual(2, s.mySqrt(4))
        self.assertEqual(2, s.mySqrt(8))
        self.assertEqual(3, s.mySqrt(9))


if __name__ == '__main__':
    unittest.main()