Educational Codeforces Round 109（C. Robot Collisions）

Posted 2021-06-08 偶尔爆零的蒟蒻

Educational Codeforces Round 109（C. Robot Collisions）

题意

数轴$0,1,2,...,n-1,n$上每个点有一个机器人，每个机器人有左右两个运动方向，碰到$x=0,n$时，会掉头。
当两个机器人恰好同时走到一个点时，两个机器人会消失，起始时间为$0$，问每个机器人消失的时间是多少，如果不会消失则时间置为$-1$。

当时时间结束了，想到了只有同位奇数或偶数位置机器人才有相撞的可能，不知道怎么处理谁和谁最先相撞，面向样例硬写了一个WA on test 2。后来才知道要用到栈，有点像括号匹配，但方向相同的也可能会撞，还用到了一点贪心思维。当时D题以为要用栈，感觉能行，但现在想想行不通，栈一般只看前面的元素，后面的元素照顾不到，栈一般是单调的单方向的考虑问题。比如，删数问题（一个数，删掉k个，使其最小）,只考虑当前元素之前的是否构成下降沿，如果栈中元素大于当前元素，则去掉栈顶。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define inf 0x3f3f3f3f
#define rush() int T;cin>>T;while(T--)
#define pb push_back
#define mp make_pair
#define ms(a,x) memset(a,x,sizeof(a))
#define yes(); cout<<"YES\\n";
#define no(); cout<<"NO\\n";
const int MAX=3e5;
struct rob{
    int id,pos,d;//序号，位置，方向
    bool operator<(rob b){//按位置排序
        return pos<b.pos;
    }
}r[MAX+5];
stack<rob> rj,ro;//奇数位置机器人放rj，偶数位置机器人放ro
int ans[MAX+5];//每个机器人的消失时间
int main(){
    rush(){
        while(!rj.empty())rj.pop();//清空，clear用不了
        while(!ro.empty())ro.pop();
        ms(ans,-1);//初始化-1，如果不会撞，那时间就是-1
        int n,m;
        cin>>n>>m;
        for(int i=0;i<n;i++){
            r[i].id=i;
            cin>>r[i].pos;
        }
        char dir;
        for(int i=0;i<n;i++){
            cin>>dir;
            if(dir=='R'){
                r[i].d=1;
            }
            else r[i].d=-1;
        }
        sort(r,r+n);
        //一共四个方向组合
        //<<,><
        for(int i=0;i<n;i++){
            if(r[i].pos&1){//奇数位置
                if(rj.empty())rj.push(r[i]);//栈空，当前机器人位置大于，栈顶机器人位置
                else if(rj.top().d==-1&&r[i].d==-1){//栈顶和当前机器人都向左
                    int temp=rj.top().pos+(r[i].pos-rj.top().pos)/2;
                    ans[rj.top().id]=ans[r[i].id]=temp;
                    rj.pop();
                }
                else if(rj.top().d==1&&r[i].d==-1){//栈顶向右，当前向左
                    int temp=(r[i].pos-rj.top().pos)/2;
                    ans[rj.top().id]=ans[r[i].id]=temp;
                    rj.pop();
                }
                else//贪心顺序，从左向右处理，如果都往右，或着位置小的往左位置大的往右，则把当前机器人入栈
                    rj.push(r[i]);
            }
            else{//偶数位置机器人，同上处理
                if(ro.empty())ro.push(r[i]);
                else if(ro.top().d==-1&&r[i].d==-1){
                    int temp=ro.top().pos+(r[i].pos-ro.top().pos)/2;
                    ans[ro.top().id]=ans[r[i].id]=temp;
                    ro.pop();
                }
                else if(ro.top().d==1&&r[i].d==-1){
                    int temp=(r[i].pos-ro.top().pos)/2;
                    ans[ro.top().id]=ans[r[i].id]=temp;
                    ro.pop();
                }
                else
                    ro.push(r[i]);
            }
        }
        //栈内剩余的
        //剩下两种方向组合
        //<>,>>
        while(rj.size()>=2){//奇数位置
            rob t2=rj.top();rj.pop();
            rob t1=rj.top();rj.pop();
            if(t2.d==1&&t1.d==1){//>>
                int temp=m-t2.pos+(t2.pos-t1.pos)/2;
                ans[t2.id]=ans[t1.id]=temp;
            }
            else if(t2.d==1&&t1.d==-1){//<>
                int temp=(2*m-(t2.pos-t1.pos))/2;//相遇时走过的路程除以总速度
                ans[t2.id]=ans[t1.id]=temp;
            }
        }
        while(ro.size()>=2){//偶数位置，同上
            rob t2=ro.top();ro.pop();
            rob t1=ro.top();ro.pop();
            if(t2.d==1&&t1.d==1){
                int temp=m-t2.pos+(t2.pos-t1.pos)/2;
                ans[t2.id]=ans[t1.id]=temp;
            }
            else if(t2.d==1&&t1.d==-1){
                int temp=(2*m-(t2.pos-t1.pos))/2;
                ans[t2.id]=ans[t1.id]=temp;
            }
        }
        for(int i=0;i<n;i++)
            cout<<ans[i]<<" ";
        cout<<endl;
    }
}