CCF 201803-2 碰撞的小球

问题描述

　　数轴上有一条长度为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

解题思路

分两种情况，一个球时的情况，两个及其以上个球时的情况。具体见注释。

注意：首先用结构体存储位置，包括index和value。对value排序，相应的index也跟着排序。最后输出的时候对index排序，相应的value也跟着排序。然后输出value即可。

代码

#include<iostream>
#include<algorithm>
using namespace std;
 
void move(int pos[], char dir[], int n, int t, int l){
    struct {
        int index, value;
    }a[105];
    
    for (int i = 0; i < n; i++){
        a[i].index = i;
        a[i].value = pos[i];
    }
    
    for (int i = 0; i < n-1; i++){
        for (int j = n-1; j >= i+1; j--){
            if (a[j].value < a[j-1].value){
                int t = a[j].value;
                a[j].value = a[j-1].value;
                a[j-1].value = t;
                t = a[j].index;
                a[j].index = a[j-1].index;
                a[j-1].index = t;
            }
        }
    }
    while (t--){
        if (n == 1){    //只有一个小球 
            if (dir[0] == 'r'){        //向右运动 
                if (a[0].value == l){    //碰到墙 
                    a[0].value--;
                    dir[0] = 'l'; 
                }
                else {    //没碰到墙 
                    a[0].value++;
                }
            } 
            else {        //向左运动 
                if (a[0].value == 0){    //碰到墙 
                    a[0].value++;
                    dir[0] = 'r'; 
                }
                else {    //没碰到墙 
                    a[0].value--;
                }
            }
        } 
        
        else {    //两个球及以上 
        
            for (int i = 0; i < n; i++){
                
                if (i == 0){ //第一个球 
                    
                    if (dir[0] == 'r'){    //1球向右 
                        if (a[0].value == a[1].value){    //碰到2球 
                            a[0].value--;    //1球向左 
                            dir[0] = 'l'; 
                            a[1].value++;    //2球向右 
                            dir[1] = 'r';
                            i++;    //2球已经移动了，跳过 
                        } 
                        else {    //没碰到2球 
                            a[0].value++;
                        }
                    }
                    else if (dir[0] == 'l'){    //1球向左 
                        if (a[0].value == 0){    //碰到墙 
                            a[0].value++;
                            dir[0] = 'r'; 
                        }
                        else {    //没碰到墙 
                            a[0].value--;
                        }
                    }
                    
                }
                
                else if (i == n-1){    //最后一个球 
                    
                    if (dir[i] == 'r'){    //最后一个球向右 
                        if  (a[i].value == l){    //碰到墙 
                            a[i].value--;
                            dir[i] = 'l'; 
                        }
                        else {    //没碰到墙 
                            a[i].value++;
                        } 
                    }
                    else if (dir[i] == 'l'){    //最后一个球向左
                            a[i].value--;    
                    //之所以不用考虑是否碰到了球，是因为如果碰到相应的处
                    //理在前一个球代码块那里。处理后会跳过最后一个球的。 
                    }
                    
                }
                
                else{    //中间球（如果有的话）
                 
                    if (dir[i] == 'r'){    //向右 
                        if (a[i].value == a[i+1].value){    //碰到球 
                            a[i].value--;    //此球向左 
                            dir[i] = 'l'; 
                            a[i+1].value++;    //另一个球向右 
                            dir[i+1] = 'r';
                            i++;    //处理后跳过另一个球 
                        } 
                        else{    //没碰到球 
                            a[i].value++;
                        }
                    }
                    else if (dir[i] == 'l'){    //向左 
                        a[i].value--;
                    //之所以不用考虑是否碰到了球，是因为如果碰到相应的处
                    //理在前一个球代码块那里。处理后会跳过该球的。 
                    }
                    
                }    //中间球（如果有的话）
                
            }    //for循环 
        }    //两球及以上 
    
    }    //while(t--) 
    
    for (int i = 0; i < n-1; i++){
        for (int j = n-1; j >= i+1; j--){
            if (a[j].index < a[j-1].index){
                int t = a[j].value;
                a[j].value = a[j-1].value;
                a[j-1].value = t;
                t = a[j].index;
                a[j].index = a[j-1].index;
                a[j-1].index = t;
            }
        }
    }
    
    for (int i = 0; i < n; i++)
    cout<< a[i].value<<" ";
    cout<<endl;
    
}

int main()
{
    int n, L, t;
    int pos[105]; 
    char dir[105];
    cin>>n>>L>>t;
    for (int i = 0; i < n; i++){
        cin>>pos[i];
        dir[i] = 'r';
    }
    move(pos, dir, n, t, L);
    return 0;
}