- 題意：一開始會有 N 顆方塊，每顆方塊自成一堆，給定兩種操作，一種操作把含有方塊 X 的一堆放到含有方塊 Y 的一堆，一種操作詢問方塊 X 下有幾顆方塊。
- 題解：帶權並查集，設 ans[i] 為 i 下方方塊數量，sz[i] 為以i 為並查集的根情況，該並查集個數。Find 較複雜，設 u 更新前的父親為 f，更新後的父親為 z(祖先)，那麼 ans 會多 f 到 z 的方塊數量，也就是 ans[f]，更新完 p[u](u 在並查集的祖先)後，一併更新 ans[u]。Union 相對之下比較單純，在 u 併入 v 的狀況，維護好 ans[u] 和 sz[v] 就好了。當要查詢方塊 X 下有幾顆方塊，要先呼叫一次 Find 更新 ans[X] 再輸出。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
using namespace std;
typedef unsigned long long ULL;
const int INF = 1e9;
const int MXN = 1e5 + 5;
const int MXV = 3e5 + 5;
const ULL MOD = 10009;
const ULL seed = 31;
cin.tie(NULL); \
cout.tie(NULL); \
ios_base::sync_with_stdio(false);
int p[MXV], sz[MXV], ans[MXV];
struct DisjointSet
{
void init(int n)
{
for (int i = 0; i < n; i++)
{
p[i] = i;
sz[i] = 1;
ans[i] = 0;
}
}
int find(int u)
{
if (u != p[u])
{
int f = p[u];
p[u] = find(p[u]);
ans[u] += ans[f];
}
return p[u];
}
void Union(int u, int v)
{
u = find(u);
v = find(v);
if (u == v)
{
return;
}
p[u] = v;
ans[u] = sz[v];
sz[v] += sz[u];
}
};
/*
Usage
DisjointSet djs; // declare
djs.init(int n); // initialize from vertex 0 to vertex n
djs.find(int u) // find the parent of vertex u
djs.Union(int u, int v) // union vertex u and v
*/
int main()
{
DisjointSet djs;
int p;
scanf("%d", &p);
djs.init(MXV);
while (p--)
{
char ch;
int x, y;
scanf(" %c %d", &ch, &x);
if (ch == 'M')
{
scanf(" %d", &y);
djs.Union(x, y);
}
else
{
djs.find(x);
printf("%d\n", ans[x]);
}
// printf("%c %d %d\n", ch, x, y);
}
}
如果你覺得這篇文章很棒，請你不吝點讚 (ﾟ∀ﾟ)
