201803-2 碰撞的小球

数据规模和约定：对于所有评测用例，1 ≤ n ≤ 100，1 ≤ t ≤ 100，2 ≤ L ≤ 1000，0 < ai < L。L为偶数。保证所有小球的初始位置互不相同且均为偶数。

#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格/秒
		// 题目告知：0 < ai < L，故此处的判断可去掉。写上主要是为后面的判断给点启示
		if (b[i].pos == L || b[i].pos == 0) {
			b[i].v = -b[i].v;  // 改变velocity方向
		}
	}

	// 对小球位置排序的原因：方便之后对小球位置的讨论（设置小球序号的作用：恢复被打乱的次序），之后碰撞的整个过程中，所有的小球相对位置保持不变
	sort(b, b + n, cmp_pos);
	// 小球每次只移动一个单位时间的长度
	while (t--) {
		/**
		不在该循环中检测中间的小球碰撞的原因：小球的移动为单位时间1格
	    最初始时小球不会在相同的位置，移动一个单位的距离后，也不会发生碰撞，
		而且只有最右边的小球是整个碰撞的开端，当最左边的小球状态改变后，才会
		链式影响左边的小球。因此不需要在该循环中每次都检测中间小球的状态，
		而是把对中间小球的判断放在边界小球的状态改变之后。
		*/
		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