POJ 3684 Physics Experiment 弹性碰撞

一、题目大意

我们有N个半径为R厘米的球，固定在距离地面高度为H的管道上，刚开始释放第一个，之后每过一秒释放一个，释放下面的球不会影响到上面的球的高度，忽略一切阻力，认为球之间的碰撞为弹性碰撞，求出T时刻，所有小球的高度。

二、解题思路

​​​​​​​

三、代码

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
int C, N, H, R, T, k;
double Hdouble, Rdouble, Tdouble, t, ans[107], g = 10.0;
void input()
{
    scanf("%d%d%d%d", &N, &H, &R, &T);
    Hdouble = H * 1.0, Rdouble = R * 0.01, Tdouble = T * 1.0;
    t = sqrt(2.0 * Hdouble / g);
}
void solve()
{
    for (int i = 1; i <= N; i++)
    {
        if ((i - 1) >= T)
        {
            ans[i] = Hdouble;
            // 还没有下落
            continue;
        }
        // 每个小球的开始时间与它的顺序有关
        double currentT = Tdouble - (i - 1) * 1.0;
        k = floor(currentT / t);
        if (k % 2 == 0)
        {
            ans[i] = Hdouble - pow(currentT - k * t, 2) * g / 2.0;
        }
        else
        {
            ans[i] = Hdouble - pow((k + 1) * t - currentT, 2) * g / 2.0;
        }
    }
}
void outputAns()
{
    sort(ans + 1, ans + 1 + N);
    for (int i = 0; i < N; i++)
    {
        ans[i + 1] += (i * 2.0 * Rdouble);
        printf("%.2f%c", ans[i + 1], i + 1 == N ? '\n' : ' ');
    }
}
int main()
{
    scanf("%d", &C);
    while (C--)
    {
        input();
        solve();
        outputAns();
    }
    return 0;
}