ccf 201803-2 碰撞的小球(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

样例说明

  初始时,三个小球的位置分别为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、碰撞边缘的只可能是第一个小球或最后一个小球(按照位置排序之后)

2、只有相邻的两个小球才有可能相撞

 1 class ball(object):
 2     def __init__(self, num, loc):
 3         self.num = num
 4         self.loc = loc
 5 
 6 
 7 n, L, t = list(input().split())
 8 n = int(n)
 9 L = int(L)
10 t = int(t)
11 temp = list(map(int, input().split()))
12 loc_list = []
13 for j in range(n):
14     loc_list.append(ball(j+1, temp[j]))
15 # 向右为1, 向左为-1
16 dir_list = [1]*n
17 loc_list.sort(key = lambda x: x.loc)
18 for i in range(t):
19     # 进行移动
20     for j in range(n):
21         loc_list[j].loc += dir_list[j]
22     # 检查是否碰撞
23     # 只有第一个和最后一个球,有碰墙的机会
24     if loc_list[0].loc <= 0: dir_list[0] = dir_list[0] * -1
25     if loc_list[n - 1].loc >= L: dir_list[n - 1] = dir_list[n - 1] * -1
26     # 只有相邻两个球有相撞的机会
27     for j in range(n-1):
28         if loc_list[j].loc == loc_list[j+1].loc:
29             dir_list[j] = dir_list[j] * -1
30             dir_list[j+1] = dir_list[j+1] * -1
31 
32 loc_list.sort(key=lambda x:x.num)
33 for i in range(len(loc_list)):
34     print(loc_list[i].loc, end="	")
原文地址:https://www.cnblogs.com/yxh-amysear/p/9596396.html