6-2 弹球距离 分数 10 作者 DS课程组 单位 浙江大学 设有一个球从高度为h米的地方落下，碰到地面后又弹到高度为原来p倍的位置，然后又落下，再弹起，再落下…。请编写函数求初始高度为h的球下落后到基本停下来（高度小于给定阈值TOL）时在空中所经过的路程总和。 函数接口定义： double dist( double h, double p ); 其中h是球的初始高度，p是球弹起高度与弹起前落下高度的比值；函数dist要返回球下落后到基本停下来时在空中所经过的路程总和。注意：当弹起的高度小于裁判程序定义的常数TOL时，弹起的距离不计算在内。 裁判测试程序样例： #include <stdio.h> #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

以下是 C++ 的代码实现：

#include <cmath>

using namespace std;

const double TOL = 1e-3;

double dist(double h, double p) {
    double s = 0;
    while (h > TOL) {
        s += h;
        h *= p;
        s += h;
    }
    s -= h;
    return s;
}

以下是 Python 的代码实现：

import math

TOL = 1e-3

def dist(h, p):
    s = 0
    while h > TOL:
        s += h
        h *= p
        s += h
    s -= h
    return s