CCF-CSP-201803-2-碰撞的小球 JAVA

// 碰撞的小球
import java.util.Scanner;

public class 碰撞的小球 {
    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[] ballXArr = new int[n];

        // 录入小球
        for (int i = 0; i < ballXArr.length ; i++) {
            ballXArr[i] = input.nextInt();
        }

        // 初始化小球集合
        /*
        *   参数1：小球个数
        *   参数2：边界
        * */
        ballList bL = new ballList(n,L);

        // 创建小球
        // 参数：所有小球的坐标数组
        bL.addBall(ballXArr);

        // 执行 t 秒
        for (int i = 0; i < t ; i++) {
            bL.nextTime();
        }

        // 输出所有小球的位置
        bL.printBallX();
    }
}

class ballList {
    ball[] bList; // 小球集合
    int ballNum = 0; // 小球总数
    int boundary; // 边界

    // 初始化小球集合
    /*
     *   参数1：小球个数
     *   参数2：边界
     * */
    public ballList(int ballSum, int boundary) {
        this.bList = new ball[ballSum];
        this.boundary = boundary;
    }

    // 创建所有球
    // 参数：所有小球的坐标数组
    public void addBall(int[] ballX) {
        for (int itemX : ballX) {
            this.bList[ballNum++] = new ball(itemX);
        }
    }

    // 所有小球进入下一秒
    public void nextTime() {
        
        // 调用所有小球的nextTime方法
        for (int i = 0; i < this.ballNum; i++) {
            this.bList[i].nextTime();
        }

        // 判断小球是否碰撞边界
        isCollision1();

        // 判断小球是否互撞
        isCollision2();
    }

    // 判断小球是否碰撞边界
    public void isCollision1() {
        for (int i = 0; i < this.ballNum; i++) {
            if (this.bList[i].x == 0 || this.bList[i].x == this.boundary) {
                bList[i].change(); // 改变方向
            }
        }
    }

    // 判断小球是否互撞
    /*
     *   1. 小球进行两两判断  判断条件为：两个小球的坐标是否一样
     *   2. 判断过的小球没必要再判断 + 小球不会跟自己判断  = 第二层循环 j = i + 1
     *   3. 第一层循环的最后一个小球已经被判断过 第一层循环 i < this.ballNum - 1
     * */
    
    public void isCollision2() {
        for (int i = 0; i < this.ballNum - 1; i++) {
            for (int j = i + 1; j < this.ballNum; j++) {
                if (this.bList[i].x == this.bList[j].x) {
                    this.bList[i].change();
                    this.bList[j].change();
                }
            }
        }
    }

    // 输出所有小球位置
    public void printBallX() {
        for (int i = 0; i < this.ballNum ; i++) {
            System.out.printf("%d ",this.bList[i].x);
        }
    }
}

class ball {
    int x; // 在数轴上的位置
    int headOrBack = 1; // 控制小球的前进或后退  1前进  -1后退

    // 初始化小球
    // 参数：小球的起始坐标
    public ball(int x) {
        this.x = x;
    }

    // 下一秒
    public void nextTime() {
        this.x += headOrBack;
    }

    // 改变方向
    public void change() {
        this.headOrBack = -this.headOrBack;
    }
}