CSP-2018-3 碰撞的小球 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

样例说明

样例输入

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

样例输出

6 6 8 2 4 0 4 12 10 2

思路

x(位移)=x0(初始位置)+v(速度)*t(时间)

所以构建记录位移的列表和记录速度的列表，时间决定了循环次数，初始态均给出。

循环：

1.判断是否碰撞：情况1：小球相遇，那么会有相同的位移量 

                            情况2：小球碰壁，那么会有位移等于最大/最小值

2.对于碰撞的小球速度反向，即乘以-1

3.刷新位移列表，使用公式x=x0+vt

代码

n, l, t = map(int, input().split())
list_dis = [i for i in map(int, input().split())]
v_flag = [1 for i in range(n)]
for k in range(t):
    if l in list_dis:
        v_flag[list_dis.index(l)] *= -1
    if 0 in list_dis:
        v_flag[list_dis.index(0)] *= -1
    for d in range(len(list_dis)-1):
        for m in range(d+1,len(list_dis)):
            if list_dis[d]==list_dis[m] :
                v_flag[d] *= -1
                v_flag[m] *= -1
    for j in range(n):
        list_dis[j] += v_flag[j]
for dis in list_dis:
    print(dis, end=' ')