csp_碰撞的小球

题目链接

使用一个结构体存储小球的位置和运动方向，每过1s就改变每个小球的位置，这可以通过一层for解决，如果到达边界就方向反向，（分析出不会有俩小球同时到边界），接着用双重for检查每个小球是否存在和它位置相同的小球，（不会同时有3小球碰撞，同一位置更不会有四个小球，因此在同一位置肯定只有一对球），改变这一对球的方向，这里有一个问题是不加处理的话，A会找到B同位置然后方向互换，B会找到A同位置然后方向互换，导致相当于没变更方向，解决方法是对每个坐标位置记录在这里碰撞的次数，初值是0，有小球在此位置碰撞就自增1，如果是奇数，表示第一次碰撞，让两球方向互换，如果偶数则不变方向。也可以用个bool类型辅助判断。
处理麻烦了，找同一位置j不需要从头开始找，从j+1就可：

//对每个球检查位置相同的，改变方向
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                if(balls[i].pos == balls[j].pos){
                    //奇数是第一次改变
                    balls[i].dir=!balls[i].dir;
                    balls[j].dir=!balls[j].dir;
                }
            }
        }

代码：

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

struct Ball{
    int pos;
    bool dir;//true为向右 false为向左
}balls[101];
int position[1001]={0};//奇数表示第一次处理碰撞情形，偶数表示第二次处理碰撞情形

int main()
{
    int n,L,t;
    cin >> n >> L >> t;
    for(int i=0;i<n;i++){
        cin >> balls[i].pos;
        balls[i].dir=true;//开始都是向右走
    }
    while(t--){
        //每一秒先依据位置和方向改变位置
        for(int i=0;i<n;i++){
            if(balls[i].dir==true) balls[i].pos++;
            else balls[i].pos--;
            if(balls[i].pos==0 || balls[i].pos==L)
                balls[i].dir=!balls[i].dir;//到边界改变位置
        }
        //对每个球检查位置相同的，改变方向
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){//处理麻烦了，其实j从i+1遍历就行
                if(i!=j && balls[i].pos == balls[j].pos){
                    position[balls[i].pos]++;
                    if(position[balls[i].pos] &1){
                        //奇数是第一次改变
                        balls[i].dir=!balls[i].dir;
                        balls[j].dir=!balls[j].dir;
                    }
                }
            }
        }
    }
    for(int i=0;i<n;i++){
        cout << balls[i].pos << " ";
    }
    return 0;
}