CCF 201803-2 碰撞的小球

#include<iostream>
using namespace std;

int main()
{
    int n,L,t;
    int a[100];
    int flag[100]= {0}; //0为向前，1为向后
    cin>>n>>L>>t;
    for(int i=0; i<n; i++)cin>>a[i];
    for(int i=0; i<t; i++)
    {
        //cout<<"1";
        for(int j=0; j<n; j++) //先计算出下一秒的位置
        {
            if(flag[j]==0)a[j]++;
            else a[j]--;
        }

        for(int i=0; i<n; i++)
        {
            if(a[i]==L)flag[i]=1;//判断球是否到达终点
            if(a[i]==0)flag[i]=0;//判断球是否到达起点
        }

        for(int k=0; k<n; k++) //开始判断相撞的球
        {
            for(int p=k+1; p<n; p++)
            {
                if(a[k]==a[p])//还要判断两个球在哪个方向
                {
                    if(flag[k]==0)
                    {
                        flag[k]=1;
                        flag[p]=0;
                    }
                    else
                    {
                        flag[k]=0;
                        flag[p]=1;
                    }

                }
            }
        }

    }
    for(int i=0; i<n; i++)
    {
        cout<<a[i]<<" ";
    }

    return 0;
}