CCF第二题之碰撞的小球

代码参考大佬碰撞的小球

#include <bits/stdc++.h>
using namespace std;
#define N 100
struct Node
{
    int pos;//位置
    int dir;//方向
}node[N];
int main()
{
    std::ios::sync_with_stdio(false);
    int n,l,t;
    cin>>n>>l>>t;
    int  id[l+1]={0};//记录线段上小球信息，为0表示没有小球，否则表示小球编号
    for(int i=1;i<=n;i++)
    {
        cin>>node[i].pos;
        node[i].dir=1;
        id[node[i].pos]=i;
    }
    while(t--)
    {
        for(int i=1;i<=n;i++)
        {
            id[node[i].pos]=0;//小球从当前位置走了，之前的位置就设为没有球,0
            node[i].pos=node[i].pos+node[i].dir;
            //更新小球的位置
            if(id[node[i].pos]!=0)//这里最难理解，举例说明！！！
            {
                node[i].dir=-node[i].dir;//当前节点的方向取反
                node[id[node[i].pos]].dir=-node[id[node[i].pos]].dir;//本身在这个位置的点的方向取反
                //一个个点的遍历，碰撞应该发生在遍历后面那个点的时候，所有之前的id[node[i].pos]才会不等于0
                //node[i]就是后面的那个点，而node[id[node[i].pos]]是上一个移到这个位置的点
            }
            else if(node[i].pos==0||node[i].pos==l)
            {//到了端点
                node[i].dir=-node[i].dir;//当前节点的方向取反
            }
            id[node[i].pos]=i;//将小球移动到当前位置
        }
    }
    for(int i=1;i<=n;i++)
    {
        cout<<node[i].pos<<" ";
    }
    return 0;

}