CCF-碰撞的小球(Java實現)
問題描述 數軸上有一條長度為L(L為偶數)的線段,左端點在原點,右端點在座標L處。有n個不計體積的小球線上段上,開始時所有的小球都處在偶數座標上,速度方向向右,速度大小為1單位長度每秒。 提示 因為所有小球的初始位置都為偶數,而且線段的長度為偶數,可以證明,不會有三個小球同時相撞,小球到達線段端點以及小球之間的碰撞時刻均為整數。 輸入格式 輸入的第一行包含三個整數n, L, t,用空格分隔,分別表示小球的個數、線段長度和你需要計算t秒之後小球的位置。 輸出格式 輸出一行包含n個整數,用空格分隔,第i個整數代表初始時刻位於ai的小球,在t秒之後的位置。 樣例輸入 3 10 5 樣例輸出 7 9 9 樣例說明 初始時,三個小球的位置分別為4, 6, 8。 三秒後,第二個小球與第三個小球在位置9發生碰撞,速度反向(注意碰撞位置不一定為偶數),三個小球位置分別為7, 9, 9。 四秒後,第一個小球與第二個小球在位置8發生碰撞,速度反向,第三個小球碰到牆壁,速度反向,三個小球位置分別為8, 8, 10。 五秒後,三個小球的位置分別為7, 9, 9。 樣例輸入 10 22 30 樣例輸出 6 6 8 2 4 0 4 12 10 2 資料規模和約定 對於所有評測用例,1 ≤ n ≤ 100,1 ≤ t ≤ 100,2 ≤ L ≤ 1000,0 < ai < L。L為偶數。 |
思路:開兩個陣列,一個記錄小球的位置,另一個標記小球的方向。在改變小球的方向過程中,採用比較下標大小的方式,這就保證了交換方向只交換一次。PS:以為會超時,但是交上去竟然過了。。。
程式碼:
import java.util.*;
public class Main {
public static void main(String[] args){
int n,L,t;
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
L = sc.nextInt();
t = sc.nextInt();
int pos[] = new int[n];
boolean flag[] = new boolean[n];
for(int i=0;i<n;i++){
pos[i] = sc.nextInt();
if (pos[i] == L)
flag[i] = false;//向左
else
flag[i] = true;//向右
}
for(int m=1;m<=t;m++){
for(int j=0;j<n;j++){
if(flag[j]==true) {
pos[j]++;
if(pos[j] == L)
flag[j] = false;
}
else{
pos[j]--;
if(pos[j] == 0)
flag[j] = true;
}
}
for(int j=0;j<n;j++){
for(int k=0;k<n;k++){
if(k==j)
continue;
if(pos[k] == pos[j]){
if(k<j){
boolean tmp = flag[k];
flag[k] = flag[j];
flag[j] = tmp;
}
}
}
}
}
for(int i=0;i<n;i++)
System.out.print(pos[i]+" ");
}
}