LeetCode 69 题

Stella981
• 阅读 517

1.题目要求

实现 int sqrt(int x) 函数。

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

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

示例 1:

输入: 4
输出: 2

示例 2:

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

2.初次尝试

这道题很明显不是让我们调用 Math.sqrt() 方法来计算,而是自己实现一个求平方根的算法。第一反应想到的方法是暴力循环求解!从 1 开始依次往后求平方数,当平方数等于 x 时,返回 i ;当平方数大于 x 时,返回 i - 1。

 1 class Solution {
 2     public int mySqrt(int x) {
 3         for(int i = 0; i <= x; i++) {
 4             if(i * i >= x){
 5                 if(i * i == x)
 6                     return i;
 7                 else
 8                     return i - 1;
 9             }
10         }
11         return 0;
12     }
13 }

这种方案通过了测试,但是成绩惨不忍睹:

执行用时 : 105 ms, 在Sqrt(x)的Java提交中击败了5.53% 的用户
内存消耗 : 33.1 MB, 在Sqrt(x)的Java提交中击败了83.60% 的用户

好奇心驱使下,我用了 Math.sqrt() 方法又提交了一遍答案:

1 class Solution {
2     public int mySqrt(int x) {
3         return (int)Math.sqrt(x);
4     }
5 }

成绩有点无解:

执行用时 : 5 ms, 在Sqrt(x)的Java提交中击败了99.24% 的用户
内存消耗 : 33.2 MB, 在Sqrt(x)的Java提交中击败了80.86% 的用户

这样一道乍看之下有点“蠢”的题目,其实有很多可以深究的地方。Math.sqrt() 用的是什么算法?求平方数的算法还有哪些?

Google 了一下“求平方根”,看到了两个出镜率最高的名词,一个是我们耳熟能详的“二分法”,另一个则是我第一次听说的“牛顿迭代法“。难得五一假期有空,决定了解一下”牛顿迭代法“并自己写出基于此算法的解题答案。

3.牛顿迭代法

我是根据知乎上一个回答了解牛顿迭代法的,链接贴出来了,有兴趣的朋友可以移步去看一下。这里简单的通过他的文章说明一下思路。

如何通俗易懂地讲解牛顿迭代法求开方?数值分析?​www.zhihu.com

这种算法的一个重要的思想是:切线是曲线的线性逼近。基于这种思想,牛顿尝试用切线来研究曲线的问题,例如用切线的根近似的求出曲线的根。然后他观察到一个现象,当在曲线上取某一点作切线时,以该切线的根作垂线,在垂线和曲线的交点处再作切线,以此循环往复,切线的根逐渐会逼近曲线的根。如图所示(A点时第一个取的点)。

LeetCode 69 题

当然,其实这种迭代并不是一定能保证会向曲线的根逼近,具体原因可以移步上述链接。但是求二次方程的根是没有问题的。

4.牛顿迭代法求平方根

回归到题目,求 a 的平方根,实际上可以转换成求二次方程 x^2 - a = 0 的解的问题。然后可以作出该二次方程的曲线,通过迭代逼近曲线 y = 0 处 x 的值,该 x 即是需要求得的答案。提现到程序中如下:

 1 class Solution {
 2     public int mySqrt(int x) {
 3         if(x == 1)
 4             return 1;
 5         double _x = x >> 1;
 6         double _y = _x * _x - x;
 7         double a = (-_y + 2 * _x * _x) / 2 / _x;
 8         while(_y > 0.1 || _y < -0.1){
 9             _x = a;
10             _y = _x * _x - x;
11             a = (-_y + 2 * _x * _x) / 2 / _x;
12         }
13         return (int)_x;
14     }
15 }

这个程序很直观的反应了迭代的过程,"_x" 是二次方程的横坐标,"_y" 是方程的纵坐标,"a" 是切线与 x 轴的交点处的横坐标。选取的第一点为 x / 2 作为横坐标,当 _y 的值逼近 0 的时候,返回 _x。该方法的成绩很接近 Math.sqrt(),结果为:

执行用时 : 6 ms, 在Sqrt(x)的Java提交中击败了92.91% 的用户
内存消耗 : 33.7 MB, 在Sqrt(x)的Java提交中击败了75.11% 的用户

5.简化

这里其实可以注意到,该二次方程一定是关于 y 轴对称的,而且二次方程在迭代过程中,若初始点在根的右边,则迭代的点会一直出现在根的右边,且一直逼近根。我们要找的其实是比根小的最大的整数,可以把 a 换成 int 类型,在逼近过程中,当 a 第一次小于等于 x / a 时,返回 a。

对程序进行简化,去掉一些不必要的参数,优化最后判断足够逼近的方式,最后程序为:

 1 class Solution {
 2     public int mySqrt(int x) {
 3         if(x == 1 || x == 0)
 4             return x;
 5         int a = x >> 1;
 6         while(a > x / a){
 7             a = (a + x/a) / 2;
 8         }
 9         return a;
10     }
11 }
点赞
收藏
评论区
推荐文章
blmius blmius
2年前
MySQL:[Err] 1292 - Incorrect datetime value: ‘0000-00-00 00:00:00‘ for column ‘CREATE_TIME‘ at row 1
文章目录问题用navicat导入数据时,报错:原因这是因为当前的MySQL不支持datetime为0的情况。解决修改sql\mode:sql\mode:SQLMode定义了MySQL应支持的SQL语法、数据校验等,这样可以更容易地在不同的环境中使用MySQL。全局s
Wesley13 Wesley13
2年前
java将前端的json数组字符串转换为列表
记录下在前端通过ajax提交了一个json数组的字符串,在后端如何转换为列表。前端数据转化与请求varcontracts{id:'1',name:'yanggb合同1'},{id:'2',name:'yanggb合同2'},{id:'3',name:'yang
皕杰报表之UUID
​在我们用皕杰报表工具设计填报报表时,如何在新增行里自动增加id呢?能新增整数排序id吗?目前可以在新增行里自动增加id,但只能用uuid函数增加UUID编码,不能新增整数排序id。uuid函数说明:获取一个UUID,可以在填报表中用来创建数据ID语法:uuid()或uuid(sep)参数说明:sep布尔值,生成的uuid中是否包含分隔符'',缺省为
Wesley13 Wesley13
2年前
MySQL的ROUND函数
ROUND(_X_)ROUND(_X_,_D_)返回参数_X_,其值接近于最近似的整数。在有两个参数的情况下,返回_X_,其值保留到小数点后_D_位,而第_D_位的保留方式为四舍五入。若要接保留_X_值小数点左边的_D_位,可将_D_设为负值。mysqlSELECTROUND(1.23);       
Wesley13 Wesley13
2年前
mysql中时间比较的实现
MySql中时间比较的实现unix\_timestamp()unix\_timestamp函数可以接受一个参数,也可以不使用参数。它的返回值是一个无符号的整数。不使用参数,它返回自1970年1月1日0时0分0秒到现在所经过的秒数,如果使用参数,参数的类型为时间类型或者时间类型的字符串表示,则是从1970010100:00:0
Stella981 Stella981
2年前
HIVE 时间操作函数
日期函数UNIX时间戳转日期函数: from\_unixtime语法:   from\_unixtime(bigint unixtime\, string format\)返回值: string说明: 转化UNIX时间戳(从19700101 00:00:00 UTC到指定时间的秒数)到当前时区的时间格式举例:hive   selec
Stella981 Stella981
2年前
Python基础之格式化输出、运算符、数字与布尔值互换以及while...else
python是一天学一点,就这样零零碎碎……格式化输出  %是占位符,%s是字符串格式,%d整数格式,%f是浮点数格式1nameinput('输入姓名')2ageinput('输入年龄')3str'年龄:%d,姓名:%s'%(int(age),name)4str
Stella981 Stella981
2年前
279. 完全平方数 leetcode JAVA
题目:给定正整数 _n_,找到若干个完全平方数(比如 1,4,9,16,...)使得它们的和等于_n_。你需要让组成和的完全平方数的个数最少。示例 1:输入:_n_12输出:3解释:12444.示例2:输入:_n_13输出:2解释:134
Stella981 Stella981
2年前
LeetCode 5561. 获取生成数组中的最大值
文章目录1\.题目2\.解题1\.题目给你一个整数n。按下述规则生成一个长度为n1的数组nums:nums00nums11当2<2i<n时,nums2inumsi
Stella981 Stella981
2年前
LeetCode:(14. 最长公共前缀!!!!!)
题目:14\.最长公共前缀编写一个函数来查找字符串数组中的最长公共前缀。如果不存在公共前缀,返回空字符串“”。示例1:输入:\“flower”,“flow”,“flight”\输出:“fl”示例2:输入:\“dog”,“racecar”,“car”\输出:“”