思路如代码注释所示:
import java.util.Arrays;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner input = new Scanner(System.in);
int n = input.nextInt();
int L = input.nextInt();
int t = input.nextInt();
int[] a = new int[n];
for (int i = 0; i < n; i++) {
a[i] = input.nextInt() + i * 1001;
//为了下面排序的时候,能找到对应的下标,a[i]/1001的商对应的就为原来小球的下标
}
//排序,让小球按照坐标的大小排序,方便了解碰撞。
for (int i = 0; i < a.length - 1; i++) {
int k = i;
for (int j = i + 1; j < a.length; j++) {
if (a[j] % 1001 < a[k] % 1001) {
k = j;
}
}
if (k != i) {
a[k] = a[i] + a[k];
a[i] = a[k] - a[i];
a[k] = a[k] - a[i];
}
}
//建立布尔数组,全部初始化为true。用于判断是前进还是后退
boolean[] flag = new boolean[n];
Arrays.fill(flag, true);
for (int i = 1; i <= t; i++) {
//判断,当小球到最右端时,为false,最左端时为true。
if (a[a.length - 1] % 1001 == L) {
flag[n - 1] = false;
} else if (a[0] % 1001 == 0) {
flag[0] = true;
}
for (int j = 1; j < n; j++) {
// 判断是否相撞,相撞的话前面一个向前走,后面一个相后走。
// !!记得要%1001;
if (a[j] % 1001 == a[j - 1] % 1001) {
flag[j] = true;
flag[j - 1] = false;
}
}
for (int j = 0; j < n; j++) {
// 如果是true则向前走,false向后走。
if (flag[j] == true) {
a[j]++;
} else {
a[j]--;
}
}
}
Arrays.sort(a);
//排序,还原原来的对应位置。(因为原来后一个位置比前一个位置大1001)
for (int j = 0; j < a.length; j++) {
System.out.printf("%d ", a[j] % 1001);
}
}
}