#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100;
class Ball {
public:
int id; // 标记
int pos;
int v; // velocity
} b[N + 10];
bool cmp_pos(Ball b1, Ball b2) {
return b1.pos < b2.pos;
}
bool cmp_id(Ball b1, Ball b2) {
return b1.id < b2.id;
}
int main() {
// freopen("in.txt", "r", stdin);
int n, L, t;
cin >> n >> L >> t;
for (int i = 0; i < n; i++) {
b[i].id = i;
cin >> b[i].pos;
b[i].v = 1; // velocity = 1格/秒
if (b[i].pos == L || b[i].pos == 0) {
b[i].v = -b[i].v; // 改变velocity方向
}
}
// 对小球位置排序的原因:方便之后对小球位置的讨论(设置小球序号的作用:恢复被打乱的次序),之后碰撞的整个过程中,所有的小球相对位置保持不变
sort(b, b + n, cmp_pos);
while (t--) {
// 小球每次只移动一个单位时间的长度
for (int i = 0; i < n; i++) {
b[i].pos += b[i].v; // s = velocity * t(t=1) <==> b[i].v * 1;
// 检测两端的小球是否和边界相撞
// 不在该循环中检测中间的小球碰撞的原因:
if (b[i].pos == L || b[i].pos == 0) {
b[i].v = -b[i].v; // 修改速度方向,确保小球下次移动时的方向
}
}
// 检测中间的小球是否碰撞
for (int i = 1; i < n; i++) {
// 由于对小球在坐标轴上的位置进行了排序,碰撞发生在相邻两球
if (b[i].pos == b[i - 1].pos) {
// 碰撞后,修改速度的方向即可
b[i].v = -b[i].v;
b[i - 1].v = -b[i - 1].v;
}
}
}
// 设置id的原因:用于将pos排序后的小球恢复到最初的状态
sort(b, b + n, cmp_id);
for (int i = 0; i < n; i++) {
cout << b[i].pos << " ";
}
return 0;
}
参考:https://blog.csdn.net/tigerisland45/article/details/81660041