POJ 1988 Cube Stacking（带权并查集）

http://poj.org/problem?id=1988

题意：

有n个砖堆，一开始每块砖都位于一个堆，现在有2种操作，操作1是将x所在的砖堆放在y所在砖堆上，操作2是计算x砖下面有多少块砖。

思路：

合并的话容易想到用并查集来维护，由于这里还需要计算每块砖下面的砖数，所以并查集需要带权值。

用ans[x]表示x下面有多少块砖，num[x]表示以x为根的子树有多少个结点。通过路径压缩就可以求出每块砖的答案。

现在看一种情况吧：

用并查集合并只之后就是

将1进行路径压缩时，它首先必须要经过2点，那么2下面的砖块数也是属于1的，所以在路径压缩的时候就可以更新1的答案值，即ans[1]+=ans[2]。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 30000+5;

int n;
int p[maxn],num[maxn],ans[maxn];

int Find(int x)
{
   if(p[x]==x)  return x;
   int tmp = p[x];
   p[x] = Find(p[x]);
   ans[x] += ans[tmp];
   return p[x];
}

int main()
{
   //freopen("in.txt","r",stdin);
   int n;
   scanf("%d",&n);
   for(int i=1;i<maxn;i++)
   {
       p[i]=i;
       num[i]=1;
       ans[i]=0;
   }
   while(n--)
   {
       char op[2]; int x, y;
       scanf("%s",op);
       if(op[0]=='M')
       {
           scanf("%d%d",&x,&y);
           int xx = Find(x);
           int yy = Find(y);
           if(xx!=yy)
           {
               p[xx]=yy;
               ans[xx]+=num[yy];
               num[yy]+=num[xx];
           }
       }
       else
       {
           scanf("%d",&x);
           Find(x);
           printf("%d
",ans[x]);
       }
   }
   return 0;
}