题目链接
http://poj.org/problem?id=1611
思路:
题意说是有n块砖,编号从1到n,有两种操作,操作一(‘M’)是把含有x编号的那一堆砖放到含有编号y的那一堆砖的上面,操作二(‘C’)是查询编号为x的砖的下面有多少块砖。传统并查集+维护size和deep两个数组,find函数在路径压缩的过程中需要把深度也依次加上,到代表这里的时候就是这个集合的最大深度,合并过程中要先将被合并的个数加到合并别人的集合的深度上去,然后在合并个数,最后将被合并的集合清零。
c++代码
#include<iostream>
#include<algorithm>
#include<stdio.h>
const int N =1e5+5;
using namespace std;
int deep[N];
int size[N];
int p[N];
int find(int x)
{
if(p[x]!=x)
{
int t=p[x];
p[x]=find(p[x]);
deep[x]+=deep[t];
}
return p[x];
}
void Union(int x,int y)
{
int xx=find(x);
int yy=find(y);
if(xx!=yy)
{
p[xx]=yy;
deep[xx]+=size[yy];
size[yy]+=size[xx];
size[xx]=0;
}
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
p[i]=i;
size[i]=1;
}
for(int i=1;i<=n;i++)
{
char op[5];
scanf("%s",op);
if(op[0]=='M')
{
int a,b;
scanf("%d %d",&a,&b);
Union(a,b);
}
else if(op[0]=='C')
{
int a;
scanf("%d",&a);
find(a);
printf("%d\n",deep[a]);
}
}
return 0;
}