Python LeetCode(69.x的平方根)

Python LeetCode(69.x的平方根)

实现 int sqrt(int x) 函数。

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

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

示例 1:

输入: 4
输出: 2

示例 2:

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

Solution:(利用牛顿迭代法求根，由于返回的是对应根的整数部分，所以当x<1时，避免让0作除数，直接返回0)

class Solution(object):
    def mySqrt(self, x):
        """
        :type x: int
        :rtype: int
        """
        val, result = 0, x
        
        if x < 1:
            return 0
        
        while abs(result - val) > 0.000001:
            val = result
            result = val - (val**2-x)/(2*val)
        
        return int(result)

solution = Solution()
print(solution.mySqrt(8))

2