CSP 201803-2碰撞的小球(JAVA代码)

问题描述
试题编号： 201803-2
试题名称： 碰撞的小球
时间限制： 1.0s
内存限制： 256.0MB
问题描述：
问题描述
　　数轴上有一条长度为L（L为偶数)的线段，左端点在原点，右端点在坐标L处。有n个不计体积的小球在线段上，开始时所有的小球都处在偶数坐标上，速度方向向右，速度大小为1单位长度每秒。
　　当小球到达线段的端点（左端点或右端点）的时候，会立即向相反的方向移动，速度大小仍然为原来大小。
　　当两个小球撞到一起的时候，两个小球会分别向与自己原来移动的方向相反的方向，以原来的速度大小继续移动。
　　现在，告诉你线段的长度L，小球数量n，以及n个小球的初始位置，请你计算t秒之后，各个小球的位置。
提示
　　因为所有小球的初始位置都为偶数，而且线段的长度为偶数，可以证明，不会有三个小球同时相撞，小球到达线段端点以及小球之间的碰撞时刻均为整数。
　　同时也可以证明两个小球发生碰撞的位置一定是整数（但不一定是偶数）。
输入格式
　　输入的第一行包含三个整数n, L, t，用空格分隔，分别表示小球的个数、线段长度和你需要计算t秒之后小球的位置。
　　第二行包含n个整数a1, a2, …, an，用空格分隔，表示初始时刻n个小球的位置。
输出格式
　　输出一行包含n个整数，用空格分隔，第i个整数代表初始时刻位于ai的小球，在t秒之后的位置。
样例输入
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 ≤ n ≤ 100，1 ≤ t ≤ 100，2 ≤ L ≤ 1000，0 < ai < L。L为偶数。
　　保证所有小球的初始位置互不相同且均为偶数。

解题思路

小球会在线段的左右端点处朝相反的方向移动,用一个boolean数组来存放每个小球的状态,true向右移动,false向左移动,int[] 数组来表示每个小球的位置
首先我们判断小球的状态,判断是否为端点,是的话就改变小球的方向
接下来判断小球是否碰撞,发生碰撞的话就改变方向
然后根据小球的状态来移动+1
不过我第一次写这个代码感觉还是挺臃肿的,后面慢慢优化吧

import java.util.Scanner;
public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int l = sc.nextInt();
		int t = sc.nextInt();
		int[] a = new int[n];
		boolean[] b = new boolean[n];
		for(int i = 0;i<n;i++) {
			a[i] = sc.nextInt();
			b[i] = true;
		}
		//向右是true,向左是false
		for(int i =0;i<t;i++) {
			for(int j=0;j<n;j++) {
				if(a[j]==l&&b[j]==true) {
					b[j]=false;
				}else if(a[j]==0&&b[j]==false) {
					b[j]=true;
				}
			}
				for(int k = 0;k<n;k++) {
					for(int p = k+1;p<n;p++) {
						if(a[k]==a[p]) {
							if(b[k]==true) {
								b[k]=false;
							}else {
								b[k]=true;
							}
							if(b[p]==true) {
								b[p]=false;
							}else {
								b[p]=true;
							}
						}
					}
				}
				for(int z=0;z<n;z++) {
				if(b[z]==true) {
					a[z]++;
				} else if(b[z]==false) {
					a[z]--;
				}
			}
			
		}
		for(int o = 0;o<n;o++) {
			System.out.print(a[o]+" ");
		}
	}	
	}