csp 201803—2 碰撞的小球（java）

思路如代码注释所示：

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);
        }

    }
}

原题链接：http://118.190.20.162/view.page?gpid=T72