python之ccf—碰撞的小球

问题描述
　　数轴上有一条长度为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

思路：提示中已经说明各个小球的初始位置都为偶数，并且不会有三个小球同时相撞，那么情况就不用考虑那么多了，这里我是使  用两个列表，一个用来保存各个小球的位置信息，另一个是用来保存对应的方向信息，当方向是向左的时候位置用-1表示，当方向是向右的时候用1表示。因为数轴上的值从左往右是增大的，所以当一个小球向右移动一下那么它的位置就是原来的位置加上对应的方向，例如：小球位置在4，方向是向右移动，那么下一秒它的位置=原来的位置4+对应的方向1，就变成5了，向左移动也是同理。

 这里有三种情况要注意更改方向：（小球一秒只移动一个长度的位置）

      1、当小球通过移动到达左端点时要立马更改该小球的方向，也就小球位置为0时要更改方向

            例：小球1位置在1，方向向左，下一秒的时候小球的位置就到达0了，这时就要更改方向，因为再下一秒的时候小球就会向右滚动了。

      2、当小球通过移动到达右端点时要立马更改该小球的方向，也就小球位置为L时要更改方向

            例：小球1位置在L-1，方向向右，下一秒的时候小球的位置就到达L了，这时就要更改方向，因为再下一秒的时候小球就会向左滚动了。

      3、当有两个小球位置相同时就证明它们两个相撞了，相撞后的下一秒这两个小球的方向就会跟原来的相反

            例如：小球1和小球2的位置都为5，小球1向右小球2向左，只有方向相反运动才会相撞嘛，这时它们撞上了，那么下一秒小球1小球2就要改变原来的方向，从原先的碰撞状态变成分开状态。

 

代码实现如下：

# 接收输入的值,各个值都是str型的
n, L, t = input().split()
# 将str型转换成int型
n, L, t = int(n), int(L), int(t)
# 输入初始位置在列表里,但是这里数据全是str型的
initialPosition = input().split()
# 方向列表，1为向右，-1为向左，开始默认全是1
direction = []
# 将初始位置的值全部转换成int型，同时给方向列表对应每个位置赋值
for i in range(initialPosition.__len__()):
    # 将初始位置信息转换成int型
    initialPosition[i] = int(initialPosition[i])
    # 将方向赋值为向右，初始方向都是向右
    direction.append(1)

# 此方法用于更新方向
def updateDirection(initialPosition1, direction, L):
    # 循环判断小球的位置是否可以进行更换方向了
    for i in range(initialPosition.__len__()):
        # 如果存在某个位置的值有两个，那么就说明有两个球进行碰撞了
        if initialPosition1.count(initialPosition1[i]) == 2:
            # 那么当前这个球的方向就要取反方向了
            direction[i] = -1*direction[i]
        # 如果某一个球的位置已经在右端点了
        if initialPosition1[i] == L:
            # 那么当前这个球的方向就要取反方向了
            direction[i] = -1*direction[i]
        # 如果某一个球的位置已经在左端点了
        if initialPosition1[i] == 0:
            # 那么当前这个球的方向就要取反方向了
            direction[i] = -1*direction[i]
    # 把修改方向后的方向列表返回
    return direction

# 这t秒内的所有操作，位置信息的改变，方向信息的改变
while t > 0:
    t -= 1
    for i in range(n):
        # 移动后的位置改变
        initialPosition[i] = initialPosition[i] + direction[i]
    # 小球的方向更新
    direction = updateDirection(initialPosition, direction, L)

# 输出经过t秒后各个小球的位置
for i in initialPosition:
    print(i, end=" ")

此过程为参考过程，希望能提供更好的解决思路，一起探讨一起学习!