参考博客:https://blog.csdn.net/u013480600/article/details/20851171
题目链接:题目链接
题意:
这里有N个龙珠,每个龙珠初始时都在本身的城市(第i个龙珠在第i个城市),然后要你回答对应的询问。
输入:首先是一个T(0< T <= 100),表示由T个实例。对于每个实例,第一行是N and Q (2 < N <= 10000 , 2 < Q <= 10000),表示接下来有Q个询问。然后对应每个询问是下面两个命令的一种:
T A B(1<= A, B <= N),表示将和A龙珠在同一个城市的所有龙珠都移到B龙珠所在的城市去。(保证A与B不在同一个城市)
Q A。你需要输出X,Y,Z。X是A龙珠目前所在的城市,Y是X城市所有的龙珠,Z是A龙珠被转移的次数。
输出:回答每条Q命令的X,Y,Z.
理解分析:(为完全理解,带加深对并查集的学习再来补分析)
#include<cstdio>
using namespace std;
const int maxn=1e4+8;
int fa[maxn];///父节点,编号为maxn的父节点
int num[maxn];///节点数,当前分量中的节点数目
int dist[maxn];///转移次数,第maxn个节点一共被转移了几次才连上当前的父节点的
int findset(int x)
{
if(fa[x]==-1) return x;
int root=findset(fa[x]);
dist[x]+=dist[fa[x]];///在寻找父节点的时候,子节点的距离要加上父节点的距离
return fa[x]=root;
}
int bind(int x,int y)
{
int fx=findset(x);
int fy=findset(y);
if(fx!=fy)
{
fa[fx]=fy;
num[fy]+=num[fx];
dist[fx]++;///子节点在往父节点移动时需要加上距离1
return 1;
}
return 0;
}
int main()
{
int n,q,T;
int case1=0;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&q);
for(int i=0;i<=n;i++)
{
fa[i]=-1;
num[i]=1;
dist[i]=0;
}
printf("Case %d:\n",++case1);
for(int i=0;i<q;i++)
{
char s;
scanf("%s",&s);
if(s=='T')
{
int x,y;
scanf("%d%d",&x,&y);
bind(x,y);
}
else
{
int x;
scanf("%d",&x);
int root=findset(x);///
int ans=num[root];
printf("%d %d %d\n",root,ans,dist[x]);
}
}
}
return 0;
}