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 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="\t")