实现 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()