POJ1988 Cube Stacking

题目链接：http://poj.org/problem?id=1988

这道题当指令是M的时候，就是并查集的合并工作了，C的时候自然就是查找，但是麻烦就是怎么去弄明白那个x下面到底有多少个元素，而且它是整列整列的弄的，所以需要找个东西标记一下。

用一个域u标记元素x的上面有多少个元素，用一个域d标记元素x的下面有多少个元素，合并的时候只要更新这两个域内的值就行了，那么对于每一个查找指令，答案就是x所在集合根节点下面的元素个数减去元素x上面的元素个数再减去x本身咯～不过这个真心不好想啊～

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 30001
struct data
{
int p, u, d;
}p
;
int find(int x)
{
int t;
if (p[x].p != x)
{
t = p[x].p;
p[x].p = find(t);
p[x].u += p[t].u;
}
return p[x].p;
}
int main()
{
int P;
int x, y;
int r1, r2;
char c;
for (int i = 1; i < N; i++)
{
p[i].p = i;
p[i].u = 0;
p[i].d = 1;
}
cin >> P;
while (P--)
{
scanf("\n%c", &c);
if (c == 'M')
{
scanf(" %d %d", &x, &y);
r1 = find(x); r2 = find(y);
if (r1 != r2)
{
p[r2].p = r1;
p[r2].u = p[r1].d;
p[r1].d += p[r2].d;
}
}
if (c == 'C')
{
scanf(" %d", &x);
y = find(x);
printf("%d\n", p[y].d - p[x].u - 1);
}
}
return 0;
}