输入:
n个小球
线段长度L
t s之后
输出:
ts之后各小球的位置
每个小球都需要至少存两个变量:位置pos,方向state
可以用±表示右左方向,用±1表示方向,修改方向时更方便。
由题意知:
1)所有小球初始位置不同且小于L,所以所有小球的初始状态state=1;
2)不会有三个小球同时碰撞,只可能两两球的位置相同;
碰撞也就是数组中存在与某个小球position的值相同的小球。
想了好久都不知道什么数据结构存小球的两个变量合适,map?list?对象?百度后,发现可以用两个数组分别存储pos和state,只要索引下标是对应即可。我吐了呀…不要总是纠结于一些复杂的数据结构…
用positions数组存储小球位置
用states数组存储小球方向
-
初始化各小球信息
-
根据上一次的state更新pos值,再根据更新后的pos值更新state
-
注意:当pos为0或者L或者pos数组中存在两个相同数值时,需要改变state为相反方向。
判断数组中是否存在两个相同值的方法也比较简单了,不赘述。
代码:
```java
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int n = scan.nextInt(); // n个小球
int L = scan.nextInt(); // 线段长度L
int t = scan.nextInt(); // ts之后
int[] positions = new int[n]; // 存储各个小球的位置
int[] states = new int[n]; // 存储各个小球的前进方向,+表示右,-表示左
// 初始化小球信息
for (int i=0;i<n;i++){
positions[i] = scan.nextInt();
states[i] = 1;
}
for (int s=1;s<=t;s++){
//先根据states数组更新positions数组
for (int i=0;i<n;i++){
if (states[i]==1){
positions[i]+=1;
}else {
positions[i]-=1;
}
}
//再根据positions数组更新states数组
for (int i=0;i<n;i++){
if (positions[i]==0 || positions[i]==L || isEqual(positions,positions[i])==true){
states[i] = -states[i];
}
}
}
for (int pos:positions){
System.out.print(pos+" ");
}
}
static boolean isEqual(int[] arr, int val){ // 判断arr中是否存在于val相等的值
// 需要先把arr中的val删除,再让val与剩下的数值比较
int flag = 0;
for (int i=0;i<arr.length;i++){
if (arr[i]==val){
flag+=1;
}
}
if (flag==2){
return true;
}
return false;
}
}