CCF 201803-2 碰撞的小球（C++）

最新推荐文章于 2022-05-31 21:34:10 发布

Brielleqqqqqqjie

最新推荐文章于 2022-05-31 21:34:10 发布

阅读量941

点赞数 3

分类专栏： CCF 文章标签： CCF

本文链接：https://blog.csdn.net/qq_41562704/article/details/95658916

版权

CCF 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

问题描述

　　数轴上有一条长度为L（L为偶数)的线段，左端点在原点，右端点在坐标L处。有n个不计体积的小球在线段上，开始时所有的小球都处在偶数坐标上，速度方向向右，速度大小为1单位长度每秒。
　　当小球到达线段的端点（左端点或右端点）的时候，会立即向相反的方向移动，速度大小仍然为原来大小。
　　当两个小球撞到一起的时候，两个小球会分别向与自己原来移动的方向相反的方向，以原来的速度大小继续移动。
　　现在，告诉你线段的长度L，小球数量n，以及n个小球的初始位置，请你计算t秒之后，各个小球的位置。

提示

　　因为所有小球的初始位置都为偶数，而且线段的长度为偶数，可以证明，不会有三个小球同时相撞，小球到达线段端点以及小球之间的碰撞时刻均为整数。
　　同时也可以证明两个小球发生碰撞的位置一定是整数（但不一定是偶数）。

输入格式

　　输入的第一行包含三个整数n, L, t，用空格分隔，分别表示小球的个数、线段长度和你需要计算t秒之后小球的位置。
　　第二行包含n个整数a1, a2, …, an，用空格分隔，表示初始时刻n个小球的位置。

输出格式

　　输出一行包含n个整数，用空格分隔，第i个整数代表初始时刻位于ai的小球，在t秒之后的位置。

样例输入

3 10 5
4 6 8

样例输出

7 9 9

样例说明

　　初始时，三个小球的位置分别为4, 6, 8。

　　一秒后，三个小球的位置分别为5, 7, 9。

　　两秒后，第三个小球碰到墙壁，速度反向，三个小球位置分别为6, 8, 10。

　　三秒后，第二个小球与第三个小球在位置9发生碰撞，速度反向（注意碰撞位置不一定为偶数），三个小球位置分别为7, 9, 9。

　　四秒后，第一个小球与第二个小球在位置8发生碰撞，速度反向，第三个小球碰到墙壁，速度反向，三个小球位置分别为8, 8, 10。

　　五秒后，三个小球的位置分别为7, 9, 9。

样例输入

10 22 30
14 12 16 6 10 2 8 20 18 4

样例输出

6 6 8 2 4 0 4 12 10 2

数据规模和约定

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

解题思路：

本题常规解法的话，不难想到设一个flag数组（1向右，-1向左）表示每个球的方向，设一个pos数组记录每个球的位置，每次更新的时候，只要将pos数组的每个值加上对应的flag处的值，如果发生碰撞，即pos数组的值到达边界，或者有两个pos在相同位置（可以用hash进行判断），则更新对应flag的值。

但是容易想到，两球碰撞的话（不包括碰撞边缘的情况），交换速度，两个球接下来走的路本来应该是对方球要走的路，其实可以看做没有碰撞，即两个球继续沿着原来的路走下去。只有边缘情况需要更新flag数组。这样做的话需要考虑另一个问题，我们弄混了每个球的标签，如何保证输出顺序呢？不难发现初始情况的位置排名决定了该球的位置排名！可以这么想，每个球的活动区域其实是限制的，假设1号球初始在2号和3号球之间，那么它始终都只能在2号和3号球之间。

如此一来设置flag数组，只有在边缘区域才更新，设置rank数组记录每个球的位置排名，最后输出的时候，将所有位置排序后，输出每号球的排名对应的pos值。

代码

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(){
	int n, L, t, cur = 0;
	scanf("%d %d %d",&n, &L, &t);
	vector<int> pos(n), init_pos(L+1, -1), flag(n,1), rank(n);
	for(int i = 0; i < n; ++ i){
		scanf("%d", &pos[i]);
		init_pos[pos[i]] = i;
	}
	for(int i = 0; i < t; ++ i){
		for(int j = 0; j < n; ++ j){
			pos[j] += flag[j];
			if(pos[j] == 0)
				flag[j] = -flag[j];
			if(pos[j] == L)
				flag[j] = -flag[j];
		}
	}
	for(int i = 0; i <= L; ++ i)
		if(init_pos[i] != -1)
			rank[init_pos[i]] = cur++;
	sort(pos.begin(), pos.end());
	printf("%d", pos[rank[0]]);
	for(int i = 1; i < n; ++ i)
		printf(" %d", pos[rank[i]]);
}

再想一想，每秒更新一次flag数组和pos数组有一丢丢麻烦，干脆直接一次性得出最终pos的值吧。

如何得出呢？

现在对每个球来讲，只要考虑他在[0,L]之间来回滚动，也就是说它的坐标变换是以T=2*L为周期进行循环的。所以所pos数组每个值加上t后，再对2*L取余，不会影响最后结果，若此时pos的值还是大于L，则最后结果应该为L - (pos[i] -L)。

代码：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(){
	int n, L, t, cur = 0;
	scanf("%d %d %d",&n, &L, &t);
	vector<int> pos(n), init_pos(L+1, -1), rank(n);
	for(int i = 0; i < n; ++ i){
		scanf("%d", &pos[i]);
		init_pos[pos[i]] = i;
	}
	t %= (L << 1);
	for(int i = 0; i < n; ++ i){
		pos[i] += t;
		pos[i] %= (L << 1);
		pos[i] = pos[i] > L ? (L << 1) - pos[i] : pos[i];
	}
	for(int i = 0; i <= L; ++ i)
		if(init_pos[i] != -1)
			rank[init_pos[i]] = cur++;
	sort(pos.begin(), pos.end());
	printf("%d", pos[rank[0]]);
	for(int i = 1; i < n; ++ i)
		printf(" %d", pos[rank[i]]);
}

结果正确，意外的是运行时间居然一样，喵喵喵？