python实现 CCF201803-2 碰撞的小球

试题编号：    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.数据结构的结构体，可以通过python类来实现

2.碰撞的小球，可以抽象为两种状态：

1）到达端点，方向改变；

2）改变后的位置若相同，则视为相撞，并且使其方向变化

二、代码实现

 

"""
1.数据结构的结构体，可以通过python类来实现
2.碰撞的小球，可以抽象为两种状态：1）到达端点，方向改变；2）改变后的位置若相同，则视为相撞，并且使其方向变化
"""
class Node():
    def __init__(self, dir, pos):
        self.dir = dir  # 球方向{1, -1}，其中1：向L方向，-1：向0方向
        self.pos = pos  # 当前球的位置

# 初始化键盘输入数据
n, L, t = list(map(int, input().strip().split(" ")))   
data = input().strip().split(" ")     
node_info = []          # 记录所有小球的位置和方向
for i in range(n):
    node_info.append(Node(1, int(data[i])))

# 小球运动t秒
for z in range(t):
    # 第z+1秒更新n个球的位置
    for i in range(n):                                        # 注意两个端点为0和L
        if node_info[i].dir == 1 and node_info[i].pos == L:   # 正向，并且到达右端点
            node_info[i].dir = -1
        elif node_info[i].dir == -1 and node_info[i].pos == 0:# 反向，并且到达左端点
            node_info[i].dir = 1
        node_info[i].pos += node_info[i].dir                  # 更新位置
    
    # 检测任意两个球是否位置相同，若相同则两个球的方向与其原来方向相反
    for k in range(n-1):
        for j in range(k + 1, n):
            if node_info[k].pos == node_info[j].pos:          # 若两球位置相同
                node_info[k].dir *= -1
                node_info[j].dir *= -1
                break      # 题中告诉只有2个小球同时相撞

# 输出结果
for i in range(n):
    if i != n-1:
        print(node_info[i].pos, end=" ")
    else:
        print(node_info[i].pos)