文章目錄
- 題目
- 程式代碼
題目
| 試題編號： | 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為偶數，
保證所有小球的初始位置互不相同且均為偶數，
程式代碼
# 輸入小球的數、線段長度和時間間隔
n,l,t = map(int,input().split())
# 輸入初始位置 創建二維串列 表示小球的位置和速度方向
ball = list(map(lambda x:([int(x),1]),input().split()))
# 添加小球序號
for i in range(n):
ball[i].append(i)
# 按小球所在位置排序
ball.sort(key=lambda x:x[0])
# 模擬t時間內小球的運動
for i in range(t):
for j in range(n):
# 模擬小球移動
if ball[j][1] == 1:
ball[j][0] += 1
else:
ball[j][0] -= 1
# 當 1 號小球運動到 0 位置時正向運動
if ball[0][0] == 0:
ball[0][1] = 1
# 當 n 號小球運動到 l 位置時反向運動
if ball[n-1][0] == l:
ball[n-1][1] = -1
# 當兩小球位置相同(相撞) 反彈
if j != 0 and ball[j-1][0] == ball[j][0] and ball[j][1] == -1:
ball[j-1][1] = -ball[j-1][1]
ball[j][1] = -ball[j][1]
j += 1 # 跳過相撞的第二個小球
# 按小球序號排序
ball.sort(key=lambda x:x[2])
# 輸出
for i in range(n):
print(ball[i][0],end=" ")
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/262040.html
標籤：python
上一篇：Python OpenCV Sobel 算子、Scharr 算子、laplacian 算子 復盤學習
下一篇：2.11生成可控的隨機資料集合