CCF 201803-2 碰撞的小球（Java 100分）

1 题目

问题描述

  数轴上有一条长度为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为偶数。
　　保证所有小球的初始位置互不相同且均为偶数。

2 我的思路

解题思路

1.构造k-v数据结构map1，约定key为小球的初始化编号，value为数组存当前小球的坐标以及小球运动的方向
2.按时间顺序模拟，按顺序移动小球的用另一个k-v(map2)记录本次所有小球的异动情况，key为坐标值，value为数组存储本次移动后在该坐标的小球集合
3.判断小球重合也就是发生碰撞（map2中各个坐标的数组是否为2）
4.按题目要求更改小球移动方向的标志位即可

3 代码

代码

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] str = br.readLine().split(" ");
		int n = Integer.parseInt(str[0]);
		int l = Integer.parseInt(str[1]);
		int t = Integer.parseInt(str[2]);
		Map<Integer, int[]> map = new HashMap<Integer, int[]>();
		String[] indexs = br.readLine().split(" ");
		for(int i=0; i<n; i++){
			int[] temp = new int[2];
			temp[0] = Integer.parseInt(indexs[i]);	//坐标
			temp[1] = 1;	//0往左，1往右	
			map.put(i+1, temp);
		}
		
		for(int i=1; i<=t; i++){	//循环时间
			Map<Integer, ArrayList<Integer>> m = new HashMap<Integer, ArrayList<Integer>>();//记录该时刻下坐标上都要哪些小球
			for(int j=1; j<n+1; j++){	//循环全部小球
				int index = 0;
				ArrayList<Integer> list = new ArrayList<Integer>();	//记录坐标上的元素标号
				//先进行移动
				if(map.get(j)[1]==1){	//向右
					index = map.get(j)[0]+1;
					map.get(j)[0] = index;
					list.add(j);
				}else if(map.get(j)[1]==0){	//向左
					index = map.get(j)[0]-1;
					map.get(j)[0] = index;
					list.add(j);
				}
				
				//记录坐标上的元素
				if(m.containsKey(index)){
					m.get(index).addAll(list);
				}else{
					m.put(index, list);
				}
				
				//判断是否到端点
				if(index==l){	//到右端点了
					map.get(j)[1] = 0;
				}else if(index==0){
					map.get(j)[1] = 1;
				}
			}
			//判断是否要变换方向
			for(Entry<Integer, ArrayList<Integer>> mm: m.entrySet()){
				if(mm.getValue().size()==2){
					ArrayList<Integer> list1 = mm.getValue();
					for(int p = 0; p<list1.size(); p++){
						if(map.get(list1.get(p))[1]==1){
							map.get(list1.get(p))[1]=0;
						}else if(map.get(list1.get(p))[1]==0){
							map.get(list1.get(p))[1]=1;
						}
					}
				}
			}
		}
		for(int i=0; i<n; i++){
			System.out.print(map.get(i+1)[0]+" ");
		}
		
		br.close();
	}
}

运行结果
10 22 30
14 12 16 6 10 2 8 20 18 4
6 6 8 2 4 0 4 12 10 2

官网评定

更多考研资料、调剂技巧可以关注我的公众号，也可以加我的微信