x 的平方根

题目描述

来源于 https://leetcode-cn.com/

实现 int sqrt(int x) 函数。

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

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

示例 1:

输入: 4
输出: 2

示例 2:

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

解法:

采用牛顿法来解,解释见 使用牛顿迭代法求平方根

import math
class Solution:
    def mySqrt(self, n):
        """
        :type x: int
        :rtype: int
        """
        if n <= 1:
            return n
        
        e = 1e-4
        
        x = n
        x_next = (x + n / x) / 2
        
        while abs(x_next - x) > e:
            x = x_next
            x_next = (x + n / x) / 2
        
        return math.floor(x)