求平方根的面试题

前几天在网上看到有人晒一道面试题,写一个函数求平方根,该函数带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;
}

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值