问题描述
试题编号: | 201803-2 |
试题名称: | 碰撞的小球 |
时间限制: | 1.0s |
内存限制: | 256.0MB |
问题描述: | 问题描述 数轴上有一条长度为L(L为偶数)的线段,左端点在原点,右端点在坐标L处。有n个不计体积的小球在线段上,开始时所有的小球都处在偶数坐标上,速度方向向右,速度大小为1单位长度每秒。 提示 因为所有小球的初始位置都为偶数,而且线段的长度为偶数,可以证明,不会有三个小球同时相撞,小球到达线段端点以及小球之间的碰撞时刻均为整数。 输入格式 输入的第一行包含三个整数n, L, t,用空格分隔,分别表示小球的个数、线段长度和你需要计算t秒之后小球的位置。 输出格式 输出一行包含n个整数,用空格分隔,第i个整数代表初始时刻位于ai的小球,在t秒之后的位置。 样例输入 3 10 5 样例输出 7 9 9 样例说明 初始时,三个小球的位置分别为4, 6, 8。 样例输入 10 22 30 样例输出 6 6 8 2 4 0 4 12 10 2 数据规模和约定 对于所有评测用例,1 ≤ n ≤ 100,1 ≤ t ≤ 100,2 ≤ L ≤ 1000,0 < ai < L。L为偶数。 |
# 输入处理 n-小球数 L-线段长度 t-时间之后 arr-小球位置数组
str1 = input().split()
n = int(str1[0])
L = int(str1[1])
t = int(str1[2])
arr = input().split()
arr = list(map(int, arr))
way = [1] * n
# 时间遍历
for i in range(t):
# 小球 边缘 调整方向
for j in range(n):
if (arr[j] == L and way[j] > 0) or (arr[j] == 0 and way[j] < 0):
way[j] *= -1
# 小球 步伐 前进一
arr = [arr[j] + way[j] for j in range(n)]
# 碰撞小球调整方向
# 获得一个位置数组 这点有不止一个球
arr_l = [j for j in range(L) if arr.count(j) > 1]
# 遍历这些位置
for node in arr_l:
# 遍历所有小球
for index, value in enumerate(arr):
# 符合位置的小球方向改变
if node == value:
way[index] *= -1
for i in range(n):
print(arr[i], end=' ')