CCF CSP 碰撞的小球 C++

碰撞的小球

代码比较长是因为看错题意，题目要求最后按序号输出位置，因此既要存序号，也要存位置。先将小球按照初始位置排序，定义另一个数组mov来记录小球移动的距离，每过一个单位时间判断是否发生碰撞，并修改mov的方向和距离。t时间后记录最后每个小球所在的位置，并按照小球序号重新排序。

#include<bits/stdc++.h>
using namespace std;
int n,l,t;
struct Ball{ //定义一个结构体，存球的序号和初始位置 
	int number;
	int location;
}ball[101];

bool cmp1(Ball a,Ball b){ //定义两个cmp函数用于将ball分别按照序号或位置排序 
	return a.location  < b.location; 
}

bool cmp2(Ball a,Ball b){
	return a.number  < b.number;
}

int mov[101][2] ={0};/*定义一个mov数组mov[n][0]用于存放每个小球相对于初始位置的行进距离， 
					mov[n][1]用于存放小球的行进方向，0代表前进，1代表后退	*/ 
int check(int w ,int x, int y,int z){
	if(x + y ==l){		//定义一个判断碰撞的函数，有碰左边，碰右边，两球相碰，不碰撞四种情况 
		return 1;
	}
	else if(x + y== w + z){
		return 2;
	}
	else if(x + y==0){
		return 3;
	}
	return 4;
}

int main(){
	
	cin>>n>>l>>t;
	
	for(int i = 0; i < n; i++){
		cin>>ball[i].location;//输入小球的初始位置，序号 
		ball[i].number = i;
	}
	sort(ball,ball+n,cmp1); //按照初始位置排序 
	for(int i = 0; i < t; i++){
		for(int k = 0; k < n; k++){
			int x = ball[k].location,y = mov[k][0],w = ball[k-1].location,z=mov[k-1][0];
			int flag = check(w,x,y,z);//检查碰撞，根据情况使小球反向 
				if(flag==1 || flag == 3){
					mov[k][1] = mov[k][1] ^ 1;//异或反向 
				}
				else if(flag == 2){
					mov[k][1] = mov[k][1] ^ 1;
					mov[k-1][1] = mov[k-1][1] ^ 1;
				}
			
		}
		for(int j = 0; j < n; j++){	//正向移动距离增加，反向减少 
			if(mov[j][1] == 0){	
				mov[j][0]++;
			}
			else if(mov[j][1] == 1){
				mov[j][0]--;
			}
		}
	/*for(int i = 0; i < n; i++){
		cout<<a[i]+mov[i][0]<<" ";
	}
	cout<<endl;*/
		
	}
	for(int i = 0; i < n; i++){
		ball[i].location+=mov[i][0]; //计算每个小球t时间后的最终位置 
	}
	sort(ball,ball+n,cmp2);//重新按照序号排序 
	for(int i = 0 ; i < n; i++){
		cout<<ball[i].location<<" "; //输出答案 
	}
	
}