[题解]AT_abc225_d [ABC225D] Play Train

题意

给定 \(N\) 个小车，每个小车的编号分别为：\(1,2,\dots,N\)。

现在有 \(Q\) 个操作，每个操作执行 \(3\) 种操作：

• 1 x y，将 \(x\) 和 \(y\) 相连。（\(y\) 在 \(x\) 之后）
• 2 x y，将 \(x\) 和 \(y\) 的连接解除。
• 3 x，输出 \(x\) 所在链的长度，及其这条链中的所有元素。（从前往后）

思路

我们可以定义一个结构体 \(a\)，对于每一个 \(a_i\) 我们保存一个 \(l\) 和 \(r\)，分别记录 \(i\) 的前一辆车的编号和 \(i\) 后一辆车的编号。

• 对于操作 1，我们只需要将 \(a_x\) 的 \(r\) 设为 \(y\)，\(a_y\) 的 \(l\) 设为 \(x\) 即可。
• 对于操作 2，我们只需要将 \(a_x\) 的 \(r\) 设为 \(0\)，\(a_y\) 的 \(l\) 设为 \(0\) 即可。
• 对于操作 3，我们可以从 \(x\) 这个点开始一直通过 \(x \leftarrow a_{x,l}\) 走到这条链的开端，再通过 \(x \leftarrow a_{x,r}\) 的方式走到链的结尾即可。（过程中需要记录一下链中的信息和长度）

Code

#include <bits/stdc++.h>  
#define re register  
  
using namespace std;  
  
const int N = 1e5 + 10;  
int n,q;  
int ans[N];  
  
struct node{  
    int l;  
    int r;  
}arr[N];  
  
inline int read(){  
    int r = 0,w = 1;  
    char c = getchar();  
    while (c < '0' || c > '9'){  
        if (c == '-') w = -1;  
        c = getchar();  
    }  
    while (c >= '0' && c <= '9'){  
        r = (r << 3) + (r << 1) + (c ^ 48);  
        c = getchar();  
    }  
    return r * w;  
}  
  
int main(){  
    n = read();  
    q = read();  
    while (q--){  
        int op;  
        op = read();  
        if (op == 1){  
            int x,y;  
            x = read();  
            y = read();  
            arr[x].r = y;  
            arr[y].l = x;  
        }  
        else if (op == 2){  
            int x,y;  
            x = read();  
            y = read();  
            arr[x].r = 0;  
            arr[y].l = 0;  
        }  
        else{  
            int x;  
            int idx = 0;  
            x = read();  
            while (arr[x].l) x = arr[x].l;  
            while (arr[x].r){  
                ans[++idx] = x;//记录   
                x = arr[x].r;  
            }  
            ans[++idx] = x;  
            printf("%d ",idx);  
            for (re int i = 1;i <= idx;i++) printf("%d ",ans[i]);  
            puts("");  
        }  
    }  
    return 0;  
}