前几天在网上看到有人晒一道面试题,写一个函数求平方根,该函数带2个参数,第一个参数是目标数字,第二个参数是精度。即:
double sqrt(const double target, const double g);
a = sqrt(t, g);
// 要求 |a - t^0.5| < g
原文看到题目,就没有接着往下看,想了一会儿,就有思路了,觉得拿来做面试题不错。可是没想到面了好几个人,都没有人能给出稍微好看点的代码。难道大家工作久了之后,写代码反而生疏了吗?一直手痒,今晚就花了几分钟实现了一下,感觉还是比较简单的。实现完成之后,感觉不错,给出的答案都满足要求了,程序应该是对了。但是后来仔细一思考,又打了些程序处理过程的trace出来,发现不对了,有bug. 什么bug呢?收敛比较慢,和原来二分查找的设想不一样。虽然也能给出结果,但是收敛慢了,程序的实现和自己的思路不一致。想了一会儿,给出了正确的代码。该sqrt函数实现大约24行左右,如下:
#include <iostream>
using namespace std;
double sqrt(const double x, const double g)
{
if (x < 0) return 0;
if (x == 1) return 1;
double low = x>1 ? 1 : x;
double high = x>1 ? x : 1;
while (true) {
double r = (low + high)/2;
cout << "low = " << low << "; high = " << high << "; r = " << r << endl;
double diff = r*r-x;
if (diff == 0) return r;
if (diff > 0) { // r > x^0.5
if ( (r-g)*(r-g) <= x ) {
return r;
}
else {
high = r;
}
}
else { // r < x^0.5
if ( (r+g)*(r+g) >= x ) {
return r;
}
else {
low = r;
}
}
}
}
int main()
{
double t = sqrt(10, 0.1);
cout << "sqrt(10) = " << t << endl;
t = sqrt(100, 0.1);
cout << "sqrt(100) = " << t << endl;
t = sqrt(0.9, 0.01);
cout << "sqrt(0.9) = " << t << endl;
return 0;
}