题目

【问题描述】数轴上有一条长度为L（L为偶数)的线段，左端点在原点，右端点在坐标L处。有n个不计体积的小球在线段上，开始时所有的小球都处在偶数坐标上，速度方向向右，速度大小为1单位长度每秒。当小球到达线段的端点（左端点或右端点）的时候，会立即向相反的方向移动，速度大小仍然为原来大小。当两个小球撞到一起的时候，两个小球会分别向与自己原来移动的方向相反的方向，以原来的速度大小继续移动。现在，告诉你线段的长度L，小球数量n，以及n个小球的初始位置，请你计算t秒之后，各个小球的位置。 提示：因为所有小球的初始位置都为偶数，而且线段的长度为偶数，可以证明，不会有三个小球同时相撞，小球到达线段端点以及小球之间的碰撞时刻均为整数。同时也可以证明两个小球发生碰撞的位置一定是整数（但不一定是偶数）。

【输入格式】输入的第一行包含三个整数n, L, t，用空格分隔，分别表示小球的个数、线段长度和你需要计算t秒之后小球的位置。第二行包含n个整数a1, a2, …, an，用空格分隔，表示初始时刻n个小球的位置。

【输出格式】输出一行包含n个整数，用空格分隔，第i个整数代表初始时刻位于ai的小球，在t秒之后的位置。

【样例输入】

3 10 5
4 6 8

【样例输出】

7 9 9

代码

#include<bits/stdc++.h>
using namespace std;

struct ball
{
    int id;
    int x;
    int dir=1; //left -1; right 1
}b[100];

bool cmp(ball A, ball B)
{
    return A.x<B.x;
}

bool cmp2(ball A, ball B)
{
    return A.id<B.id;
}

int main()
{
    int n,L,t;
    int foreball;
    cin>>n>>L>>t;
    for(int i=0;i<n;i++)
    {
        b[i].id=i;
        cin>>b[i].x;
        if(b[i].x==L)
            b[i].dir=-1;
    }

    sort(b,b+n,cmp);
    while(t--)
    {
        foreball=-1;
        for(int i=0;i<n;i++)
        {
            b[i].x+=b[i].dir;
            if(b[i].x==Lb[i].x==0)
                b[i].dir=-b[i].dir;
            else if(b[i].x==foreball)
            {
                b[i].dir=-b[i].dir;
                b[i-1].dir=-b[i-1].dir;
            }
            foreball=b[i].x;
        }
    }
    sort(b,b+n,cmp2);
    for(int i=0;i<n;i++)
    {
        cout<<b[i].x<<" ";
    }

    return 0;
}