csp 201803—2 碰撞的小球(java)

在这里插入图片描述
在这里插入图片描述

思路如代码注释所示:


import java.util.Arrays;
import java.util.Scanner;

public class Main {
    
    
    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[] a = new int[n];
        for (int i = 0; i < n; i++) {
    
    
            a[i] = input.nextInt() + i * 1001;
            //为了下面排序的时候,能找到对应的下标,a[i]/1001的商对应的就为原来小球的下标
        }
        //排序,让小球按照坐标的大小排序,方便了解碰撞。
        for (int i = 0; i < a.length - 1; i++) {
    
    
            int k = i;
            for (int j = i + 1; j < a.length; j++) {
    
    
                if (a[j] % 1001 < a[k] % 1001) {
    
    
                    k = j;
                }
            }
            if (k != i) {
    
    
                a[k] = a[i] + a[k];
                a[i] = a[k] - a[i];
                a[k] = a[k] - a[i];
            }
        }
        //建立布尔数组,全部初始化为true。用于判断是前进还是后退
        boolean[] flag = new boolean[n];
        Arrays.fill(flag, true);

       
        for (int i = 1; i <= t; i++) {
    
    
        //判断,当小球到最右端时,为false,最左端时为true。
            if (a[a.length - 1] % 1001 == L) {
    
    
                flag[n - 1] = false;
            } else if (a[0] % 1001 == 0) {
    
    
                flag[0] = true;
            }
            
            for (int j = 1; j < n; j++) {
    
    
            // 判断是否相撞,相撞的话前面一个向前走,后面一个相后走。
            // !!记得要%1001;
                if (a[j] % 1001 == a[j - 1] % 1001) {
    
    
                    flag[j] = true;
                    flag[j - 1] = false;
                }
            }

            for (int j = 0; j < n; j++) {
    
    
            // 如果是true则向前走,false向后走。
                if (flag[j] == true) {
    
    
                    a[j]++;
                } else {
    
    
                    a[j]--;
                }
            }
        }

        Arrays.sort(a);
        //排序,还原原来的对应位置。(因为原来后一个位置比前一个位置大1001)
        for (int j = 0; j < a.length; j++) {
    
    
            System.out.printf("%d ", a[j] % 1001);
        }

    }
}

原题链接:http://118.190.20.162/view.page?gpid=T72

猜你喜欢

转载自blog.csdn.net/weixin_46157208/article/details/108057416