[题解]AT_abc279_f [ABC279F] BOX

思路

对于操作 1，我们可以看做将 $x$ 号集合与 $y$ 号集合合并，并将原本的 $y$ 号集合重新开一个并查集维护。

对于操作 2，我们可以开一个数组维护每一个球所在哪一个并查集中。

对于操作 3，在操作 1 中维护每一个并查集对于了哪一个箱子即可。

因为 $n + q \leq 6 \times 10^5$，所以数组开 $6 \times 10^5$ 即可。

Code

#include <bits/stdc++.h>  
#define re register  
  
using namespace std;  
  
const int N = 6e5 + 10;  
int n,q,idx,num;  
int f[N],bd[N],nd[N],box[N];  
/*  
f 维护并查集  
bd 维护每一个箱子对应的并查集  
nd 维护每一个球对应的并查集  
box 维护每一个并查集对于的箱子   
*/   
  
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 << 1) + (r << 3) + (c ^ 48);  
        c = getchar();  
    }  
    return r * w;  
}  
  
inline int find(int x){  
    if (f[x] != x) return f[x] = find(f[x]);  
    return f[x];  
}  
  
int main(){  
    n = idx = num = read();  
    q = read();  
    for (re int i = 1;i <= n;i++) f[i] = bd[i] = nd[i] = box[i] = i;  
    while (q--){  
        int op;  
        op = read();  
        if (op == 1){  
            int x,y;  
            x = read();  
            y = read();  
            f[bd[y]] = find(bd[x]);  
            bd[y] = ++idx;  
            f[idx] = idx;  
            box[idx] = y;  
        }  
        else if (op == 2){  
            int x;  
            x = read();  
            nd[++num] = bd[x];  
        }  
        else{  
            int x;  
            x = read();  
            printf("%d\n",box[find(nd[x])]);  
        }  
    }  
    return 0;  
}