Python入门习题（41）——CCF CSP认证考试真题：碰撞的小球

问题描述

试题编号： 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为偶数。
　　保证所有小球的初始位置互不相同且均为偶数。

解题思路

1. 输入每个球的起始位置，接着所有球按位置从左到右排序。此后，无论怎么移动，有：（1）任意的相邻两球，彼此之间的位置关系不变，也即左边的球始终在左边，右边的球始终在右边。（2）球之间的碰撞只会发生在相邻两球之间，不相邻的球之间不会发生碰撞。（3）只有最左边的球会碰到左端点，最右边的球会碰到右端点。

2. 每过一秒，执行以下动作：
   （1）更新球的位置；
   （2）检查最左边的球是否碰到左端点，若是则更新最左边球的速度。
   （3）检查最右边的球是否碰到左端点，若是则更新最右边球的速度。
   （4）检查相邻球之间是否发生碰撞，若是则更新两个球的速度。

3. 按输入顺序重新排列各个球，输出它们的位置。输入阶段，为每个球保存两项信息：球的位置，球的编号。按球的编号排序，就是按输入顺序排列各个球。

参考答案

n, len, t = [int(s) for s in input().split()]
loc_index = [[int(loc_str), i] for i, loc_str in enumerate(input().split())]
     #每个球有两项信息：球的位置，球的编号（输入序列a1, a2, ..., an中的序号，从0开始）
loc_index.sort()  #按球的位置从左到右排序
# print(loc_index)

vs = [1] * n  #vs存储各个球的速度

for next_second in range(1, t + 1):  #每过一秒：
    for i in range(n):
        loc_index[i][0] += vs[i]  #更新球的位置
    if loc_index[0][0] == 0:  #碰到左端点
        vs[0] = -vs[0]
    if loc_index[n - 1][0] == len:  #碰到右端点
        vs[n - 1] = -vs[n - 1]
    #检查两个球碰撞
    for i in range(n - 1):
        if loc_index[i][0] == loc_index[i + 1][0]:
            vs[i] = - vs[i]
            vs[i + 1] = -vs[i + 1]

loc_index.sort(key = lambda item: item[1])  #按球编号排序
for loc in loc_index:
    print(loc[0], end = ' ')

测试用例

1. 题目描述给出的第一组测试用例覆盖了右端点折返和两球碰撞的情形。
2. 题目描述给出的第二组测试用例覆盖了左端点折返、右端点折返和两球碰撞的情形。
3. n=1的边界情形。
   样例输入
   1 10 3
   4
   样例输出
   7
4. t=1的边界情形。
   样例输入
   3 10 1
   4 6 8
   样例输出
   5 7 9
5. L=2的边界情形。
   样例输入
   1 2 3
   0
   **样例输出
   1

小结

1. 解答本题的关键点是：按位置对所有球排序。这简化了球之间碰撞的检测。
2. loc_index列表存储的是各个球的位置和编号。最后阶段，按编号从小到大输出。
3. 这是一道模拟题，用代码模拟了球的位置和速度的变化过程。每过一秒更新球的位置和速度。