题解：P4940 Portal2

cdx123456 · 2024-10-14 14:38:26 · 题解

其他大佬用的方法太高级了，蒟蒻看不太懂，遂写一发双向链表的题解。

注意到除了 move 操作之外都可以用栈来解决，但是处理 move 就是 O(n) 的，而栈的结构很难优化，所以放弃直接模拟的想法。

又注意到元素进行 move 操作后，栈之间元素的相邻关系没有改变，故考虑链表，然而常规链表还需要转换方向，原因见下图。

这样依次改变方向复杂度仍然为 O(n)，考虑优化。

可以仅保存元素之间的相邻关系，可以类比一下双向边，即这样。

至于插入删除之类的操作跟双向链表是一样的，注意不能写成 pre 或 nxt 之类的形式，因为这不是两个方向而是没有方向。

输出答案的时候从栈顶位置遍历到栈底位置即可。

Code

#include<bits/stdc++.h>
using namespace std;
#define int long long
int tot,a[2000010],head[2],tail[2],son[2000010][2];
void add(int x,int y){
    a[++tot]=y;
    if(son[tail[x]][0]) son[tail[x]][1]=tot;
    else son[tail[x]][0]=tot;
    son[tot][0]=tail[x];
    tail[x]=tot;
}
void del(int x){
    if(tail[x]==head[x]) return;
    int y=tail[x];
    if(son[y][0]){
        int z=son[y][0];
        if(son[z][0]==y) son[z][0]=0;
        else son[z][1]=0;
        tail[x]=z;
        son[y][0]=0;
    }else{
        int z=son[y][1];
        if(son[z][0]==y) son[z][0]=0;
        else son[z][1]=0;
        tail[x]=z;
        son[y][1]=0;
    }
}
void merge(int x,int y){
    if(head[y]==tail[y]) return;
    if(son[tail[x]][0]) son[tail[x]][1]=tail[y];
    else son[tail[x]][0]=tail[y];
    if(son[tail[y]][0]) son[tail[y]][1]=tail[x];
    else son[tail[y]][0]=tail[x];
    if(son[head[y]][0]) tail[x]=son[head[y]][0];
    else tail[x]=son[head[y]][1];
    if(son[tail[x]][0]==head[y]) son[tail[x]][0]=0;
    else son[tail[x]][1]=0;
}
bool check(int x){
    return head[x]==tail[x];
}
signed main(){
    string s;
    int x,y;
    head[0]=tail[0]=++tot;
    head[1]=tail[1]=++tot;
    while(cin>>s){
        if(s=="END"){
            cout<<"SUCCESS"<<endl;
            if(check(0)) cout<<"NONE";
            x=tail[0];
            y=0;
            while(x!=head[0]){
                cout<<a[x]<<' ';
                if(son[x][0]==y){
                    y=x;
                    x=son[x][1];
                }else{
                    y=x;
                    x=son[x][0];
                }
            }
            cout<<endl;
            if(check(1)) cout<<"NONE";
            x=tail[1];
            y=0;
            while(x!=head[1]){
                cout<<a[x]<<' ';
                if(son[x][0]==y){
                    y=x;
                    x=son[x][1];
                }else{
                    y=x;
                    x=son[x][0];
                }
            }
            cout<<endl;
            break;
        }
        if(s=="PUSH"){
            cin>>x>>y;
            add(x,y);
            cout<<"SUCCESS"<<endl;
            continue;
        }
        if(s[0]=='P'){
            cin>>x;
            if(check(x)){
                cout<<"UNSUCCESS"<<endl;
                continue;
            }
            del(x);
        }
        if(s[0]=='A'){
            cin>>x;
            if(check(0) || check(1)){
                cout<<"UNSUCCESS"<<endl;
                continue;
            }
            int y=a[tail[0]],z=a[tail[1]];
            del(0); del(1);
            add(x,y+z);
        }
        if(s=="SUB"){
            cin>>x;
            if(check(0) || check(1)){
                cout<<"UNSUCCESS"<<endl;
                continue;
            }
            int y=a[tail[0]],z=a[tail[1]];
            del(0); del(1);
            add(x,abs(y-z));
            cout<<"SUCCESS"<<endl;
            continue;
        }
        if(s[0]=='D'){
            cin>>x;
            tail[x]=head[x];
            son[head[x]][0]=son[head[x]][1]=0;
        }
        if(s[0]=='M'){
            cin>>x>>y;
            merge(x,y);
            head[y]=tail[y]=++tot;
        }
        if(s[0]=='S'){
            swap(head[0],head[1]);
            swap(tail[0],tail[1]);
        }
        cout<<"SUCCESS"<<endl;
    }
    return 0;
}