1 /*
2 * poj-1988 Cube Stacking.cpp
3 *
4 * Created on: 2012-2-12
5 * Author: LongDou
6 *
7 * 并查集：
8 * 每个节点有3个域：1、fa[x]：并查集中x的父节点 （fa[x]必压在x之下，但不一定直接相邻）
9 * 2、rank[x]：在x之下（不包括x）且在fa[x]之上(包括fa[x])的方块数
10 * （如果直接记录x之下的总方块数，则每次合并都需更新某个集合中所有元素的
11 * rank域；而这样记录的话，合并操作只需更新某个集合的根结点的rank域，
12 * 同时对一次询问“C x”，只需计算x到根的rank值之和，复杂度不大）
13 * 3、stackNum[x]：x当前所在的stack标号(仅树根节点有意义)
14 *
15 * 每次 M x y 操作， 合并x和y两个集合，注意更新各stack当前所含的方块数
16 * findSet操作在路径压缩时注意更新rank域
17 *
(另：刘汝佳的黑书中P81例题3与此题类似，解题也一样，只是略简洁一些。。。)
18 */
19 #include <cstdio>
20 using namespace std;
21
22 const int maxn = 30000 + 10;
23 const int maxp = 100000 + 10;
24
25 int num[maxn] = {}; //stack当前所含的方块数
26 int fa[maxn], rank[maxn], stackNum[maxn];
27
28 void init(){
29 for(int i=0; i<maxn; i++){
30 num[i] = 1;
31 fa[i] = i;
32 rank[i] = 0;
33 stackNum[i] = i;
34 }
35 }
36
37 int findSet(int x){
38 if(x == fa[x])
39 return x;
40
41 int tmp = fa[x];
42 fa[x] = findSet(fa[x]);
43 rank[x] += rank[tmp]; //更新rank域
44
45 return fa[x];
46 }
47
48 void unionSet(int x, int y){
49 int fx = findSet(x);
50 int fy = findSet(y);
51
52 if(fx == fy) return;
53
54 fa[fx] = fy;
55 rank[fx] += num[stackNum[fy]];
56
57 //更新各stack当前所含的方块数
58 num[stackNum[fy]] += num[stackNum[fx]];
59 num[stackNum[fx]] = 0;
60 }
61
62 int main(){
63 int p, x, y;
64 char op;
65
66 init();
67
68 scanf("%d", &p);
69 for(int i=0; i<p; i++){
70 getchar();
71 scanf("%c", &op);
72
73 if(op == 'M'){
74 scanf("%d%d", &x, &y);
75 unionSet(x, y);
76 }
77 else{
78 scanf("%d", &x);
79 int fx = findSet(x);
80 printf("%d\n", rank[x] + rank[fx]);
81 }
82 }
83
84
85
86 return 0;
87 }