碰撞的小球
代码比较长是因为看错题意,题目要求最后按序号输出位置,因此既要存序号,也要存位置。先将小球按照初始位置排序,定义另一个数组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<<" "; //输出答案
}
}