ccf 201803-2 碰撞的小球(Python)
问题描述
数轴上有一条长度为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为偶数。
保证所有小球的初始位置互不相同且均为偶数
思路
1、碰撞边缘的只可能是第一个小球或最后一个小球（按照位置排序之后）
2、只有相邻的两个小球才有可能相撞
1 class ball(object): 2 def __init__(self, num, loc): 3 self.num = num 4 self.loc = loc 5 6 7 n, L, t = list(input().split()) 8 n = int(n) 9 L = int(L) 10 t = int(t) 11 temp = list(map(int, input().split())) 12 loc_list = [] 13 for j in range(n): 14 loc_list.append(ball(j+1, temp[j])) 15 # 向右为1， 向左为-1 16 dir_list = [1]*n 17 loc_list.sort(key = lambda x: x.loc) 18 for i in range(t): 19 # 进行移动 20 for j in range(n): 21 loc_list[j].loc += dir_list[j] 22 # 检查是否碰撞 23 # 只有第一个和最后一个球，有碰墙的机会 24 if loc_list[0].loc <= 0: dir_list[0] = dir_list[0] * -1 25 if loc_list[n - 1].loc >= L: dir_list[n - 1] = dir_list[n - 1] * -1 26 # 只有相邻两个球有相撞的机会 27 for j in range(n-1): 28 if loc_list[j].loc == loc_list[j+1].loc: 29 dir_list[j] = dir_list[j] * -1 30 dir_list[j+1] = dir_list[j+1] * -1 31 32 loc_list.sort(key=lambda x:x.num) 33 for i in range(len(loc_list)): 34 print(loc_list[i].loc, end="\t")
