201803-2 碰撞的小球（sort，自定义排序）

 

试题编号：	201803-2
试题名称：	碰撞的小球
时间限制：	1.0s
内存限制：	256.0MB
问题描述：	问题描述 　　数轴上有一条长度为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为偶数。 　　保证所有小球的初始位置互不相同且均为偶数。

首先思路是判断每一步的位置，输出最后的位置，每个小球只能和左右的小球和边界有可能碰撞，那么只需按照从左到右的顺序依次判断就好，而小球相撞会反弹，则小球间的顺序不会变。

但是小球输入的顺序不是按照位置输入的，因此需要进行排序，而输出小球位置的时候要求按照输入时的顺序输出，因此需要保存每个小球的输入顺序。

#include <map>  
#include <cmath>  
#include <queue>  
#include <cstdio>  
#include <string>  
#include <cstring>  
#include <iostream>  
#include <algorithm> 
#include <sstream> 
#include <time.h> 
#include <vector>
#include <list>

using namespace std;

//201803-2 碰撞的小球
struct Ball {
	int Num, Pos;			//输入次序  当前所在位置
	int Dir;				//前进方向 1: ->  -1: <-
	Ball(int num, int pos, int dir)
	{
		Num = num; Pos = pos; Dir = dir;
	}
};

bool Cmp(Ball x, Ball y)
{
	return x.Pos < y.Pos;
}
int main()
{
	int i, j, N, L, T;
	vector<Ball>Data;
	cin >> N >> L >> T;

	int position;
	for (i = 0; i < N; i++)
	{
		cin >> position;
		Ball B(i, position, 1);
		Data.push_back(B);
	}
	sort(&Data[0], &Data[N - 1] + 1, Cmp);		//按照位置排序
	//for (i = 0; i < N; i++)
	//	cout << Data[i].Pos << endl;
	for (i = 0; i < T; i++)
	{
		Data[0].Pos += Data[0].Dir;
		if (Data[0].Pos == 0)		//第一个到边界
			Data[0].Dir = 1;
		for (j = 1; j < N; j++)
		{
			Data[j].Pos += Data[j].Dir;
			if (Data[j].Pos == Data[j - 1].Pos)
			{
				Data[j].Dir = -Data[j].Dir;
				Data[j - 1].Dir = -Data[j - 1].Dir;
			}
		}
		if (Data[N - 1].Pos == L)	//最后一个到边界
			Data[N - 1].Dir = -1;
	}
	vector<int>Out(N);
	for (i = 0; i < N; i++)
	{
		Out[Data[i].Num] = Data[i].Pos;
	}
	for (i = 0; i < N; i++)
	{
		cout << Out[i] << " ";
	}

	cin >> N;
	return 0;
}