CCF Python題解(100分)201803-2 碰撞的小球
阿新 • • 發佈:2018-11-13
CCF Python題解(100分)201803-2 碰撞的小球
from collections import defaultdict
n, L, t = map(int,input().split())
positions = map(int,input().split())
list1 = [1 for i in range(n)]
for j in range(t):
dict2 = defaultdict(list)
for i, item in enumerate(positions):
dict2[item].append(i)
for key, l in dict2.items():
if len(l) == 2:
list1[l[0]] = -list1[l[0]]
list1[l[1]] = -list1[l[1]]
for index in range(n):
if positions[index] == L or positions[index] == 0:
list1[index] = -list1[index]
positions[index] += list1[index]
for i in positions:
print(i, end=' ')