CCF 201803-2 碰撞的小球 python 满分

题目叙述

问题描述：略

输入格式：略

输出格式：略

样例

样例输入
3 10 5
4 6 8
样例输出
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. 涉及3个重点：
   （1） 到左端点换方向；
   （2）到右端点换方向；
   （3） 两点相遇换方向；

2. 最后，做一个列表加和函数。

满分代码

def ass(a, b):sum = [0] * len(a)for i in range(len(a)):sum[i] = a[i] + b[i]return sumn, lc, t = map(int, input().split())
ll = list(map(int, input().split()))
flag = [1] * n
for _ in range(t):if lc in ll:flag[ll.index(lc)] = -flag[ll.index(lc)]if 0 in ll:flag[ll.index(0)] = -flag[ll.index(0)]if len(ll) != len(list(set(ll))):templ = ll.copy()qcl=sorted(list(set(ll)))for i in qcl:if ll.count(i) == 2:flag[templ.index(i)] = -flag[templ.index(i)]templ[templ.index(i)] = -1flag[templ.index(i)] = -flag[templ.index(i)]ll = ass(ll, flag)
for i in range(n):print(ll[i], end=" ")

满分代码优化

把函数优化到里面

n, lc, t = map(int, input().split())
ll = list(map(int, input().split()))
flag = [1] * n
for _ in range(t):if lc in ll:flag[ll.index(lc)] = -flag[ll.index(lc)]if 0 in ll:flag[ll.index(0)] = -flag[ll.index(0)]if len(ll) != len(list(set(ll))):templ = ll.copy()qcl = sorted(list(set(ll)))for i in qcl:if ll.count(i) == 2:flag[templ.index(i)] = -flag[templ.index(i)]templ[templ.index(i)] = -1flag[templ.index(i)] = -flag[templ.index(i)]ll = [ll[i]+flag[i] for i in range(len(ll))]for i in range(n):print(ll[i], end=" ")