CCF 201803-2碰撞的小球(python)

碰撞的小球

借鉴其他人的思路，这个思路很好用。
本题的主要解题思路如下：
①将整个大问题分成几个小问题，分别有三种情况：球与球碰撞、球与墙碰撞、球每秒运动一个单位
②球与球碰撞情况：利用两层循环，遍历每颗球与其他球是否碰撞，若碰撞则判断当前球的方向，使其改变运动方向
③球与墙碰撞情况：遍历每颗球，判断当前的单位值是否大于线段长度或小于0，则相应的改变运动方向
④球每秒运动一个单位：遍历每颗球，根据运动方向对应的增加一个单位或减少一个单位
⑤先改变方向再运动，所以先调用两个改变方向的方法，再调用球运动的方法，最后输出运动结果即可

# 球与球碰撞后的方向
def bumpBall(p, isRL):
    for i in range(len(p)):
        for j in range(i + 1, len(p)):  # 注意这里的是从i+1开始
            if p[i] == p[j]:
                if isRL[i] == 'L':
                    isRL[i] = 'R'
                    isRL[j] = 'L'
                else:
                    isRL[i] = 'L'
                    isRL[j] = 'R'


# 球与墙面进行了碰撞后：
def bumpWall(p, isRL, L):
    for i in range(len(p)):
        if p[i] == L:
            isRL[i] = 'L'
        elif p[i] == 0:
            isRL[i] = 'R'


# 移动小球
def moveBall(p, isRL):
    for i in range(len(p)):
        if isRL[i] == 'R':
            p[i] += 1
        elif isRL[i] == 'L':
            p[i] -= 1


n, L, t = map(int, input().split())  # n小球的个数，L线段的长度，t运动的时间
p = list(map(int, input().split()))
isRL = []
for i in range(n):
    isRL.append('R')

for i in range(t):
    # 先判别方向，在进行位置的判别
    # 判断球碰球后的方向
    bumpBall(p, isRL)
    # 判断球碰墙的方向
    bumpWall(p, isRL, L)
    # 进行球的位置的判断
    moveBall(p, isRL)
for i in range(len(p)):
    print(p[i], end=' ')