PTA 6-1 弹球距离

设有一个球从高度为h米的地方落下，碰到地面后又弹到高度为原来p倍的位置，然后又落下，再弹起，再落下…。请编写函数求初始高度为h的球下落后到基本停下来（高度小于给定阈值TOL）时在空中所经过的路程总和

函数接口定义：

double dist( double h, double p );

其中h是球的初始高度，p是球弹起高度与弹起前落下高度的比值；函数dist要返回球下落后到基本停下来时在空中所经过的路程总和。注意：当弹起的高度小于裁判程序定义的常数TOL时，弹起的距离不计算在内

裁判测试程序样例：

#include 
#define TOL 1E-3

double dist( double h, double p );

int main()
{
    double h, p, d;
    scanf("%lf %lf", &h, &p);
    d = dist(h, p);
    printf("%.6f\n", d);
    return 0;
}

/* 你的代码将被嵌在这里 */

输入样例：

1.0 0.4

输出样例

2.331149

代码长度限制                                                        16 KB
时间限制                                                            400 ms
内存限制                                                            64 MB

代码实现部分

double dist( double h, double p ){
    /*初始化和  第一次是掉下去,所以第一次的高度只算落下的*/
    double sum=h;
    /*从第二次开始,都是上去+落下  即h*2的高度*/
    double temp = h*p;
    while(temp>TOL){//停止循环条件 如题目要求
        sum+=temp*2;
        temp*=p;
    }
    /*返回最后结果*/
    return sum;
}