題意:有兩種操作:一個是M,a,b;表示a集合所有的元素,置於b集合之上;
二是C ,a;表示a集合下的元素的個數有多少個;
此題;要構造兩個數組,一個是rank【i】,他記錄的是i集合元素的總個數;另一個是under【i】,記錄的是i元素下的個數,這個集合要更新,在並查集裏進行更新 ,他會從k一直更新到他的父節點,這樣其中的所有的點便都有了因隨時操作所引起的元素個數的變化;
在合併的時候,都合併到第二個集合上;這樣保證了這個集合的父節點正好位於最下邊,則有under【m】=rank【n】;
在操作c是一定要再次更新一次即find(m);保證結果的正確性;
#include"stdio.h"
#include"string.h"
int pre[100006],rank[100006];
int under[100006];
int find(int k)//合併的同時,也更新了under;
{
int temp;
if(k!=pre[k])
{
temp=find(pre[k]);
under[k]+=under[pre[k]];
pre[k]=find(pre[k]);
}
return pre[k];
}
int main()
{
int m,n,i,k;
char a[10];
while(scanf("%d",&k)!=EOF)
{
for(i=0;i<=k;i++)
{
pre[i]=i;
rank[i]=1;
under[i]=0;
}
while(k--)
{
scanf("%s",a);
if(a[0]=='M')
{
scanf("%d%d",&m,&n);
m=find(m);
n=find(n);
if(m!=n)
{
pre[m]=n;
under[m]=rank[n];
rank[n]+=rank[m];
}
}
else
{
scanf("%d",&m);
find(m);
printf("%d\n",under[m]);
}
}
}
return 0;
}