201803-2 碰撞的小球

#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

标签：Ball,int,碰撞,小球,pos,201803,id
来源： https://blog.csdn.net/qq_40750972/article/details/114554144