如题,现在有一个并查集,你需要完成合并和查询操作。
输入输出格式
输入格式:
第一行包含两个整数N、M,表示共有N个元素和M个操作。
接下来M行,每行包含三个整数Zi、Xi、Yi
当Zi=1时,将Xi与Yi所在的集合合并
当Zi=2时,输出Xi与Yi是否在同一集合内,是的话输出Y;否则话输出N
输出格式:
如上,对于每一个Zi=2的操作,都有一行输出,每行包含一个大写字母,为Y或者N
思路:
就是找共同祖先。不同祖先的就合并在一起
#include<iostream>
#include<cstdio>
using namespace std;
/*
int u[120],v[120], w[120];
int first[120];
int nex[120];
int n, m;
int main()
{
cout<<"顶点数"<<endl;
cin>>n;
cout<<"边数"<<endl;
cin>>m;
for(int i = 1; i <= n; i++)
{
first[i] = -1;
//nex[i] = -1;
}
for(int i = 1; i <= m; i++)
{
cin>>u[i]>>v[i]>>w[i];
nex[i] = first[u[i]];
first[u[i]] = i;
}
cout<<"输入以那一顶点遍历其所有的边"<<endl;
int vex;
cin>>vex;
int k = first[vex];
while(k != -1 )
{
cout<<u[k]<<" "<<v[k]<<" "<<w[k]<<endl;
k = nex[k];
}
system("pause");
}
*/
int f[10005];
int z,x,y;
int find(int a)
{
if(a == f[a])
return a;
else
return find(f[a]);//此处有个小优化,将这个return改回return f[a] = find(f[a])
//就是记忆化搜索,保存a的祖先的结果。下次搜索的时候不用继续递归去查找其祖先。
}
int main()
{
int n,m;
//cin>>n>>m;
scanf("%d%d", &n, &m);
for(int i = 0; i <= n; i++)
{
f[i] = i;
}
for(int i = 1; i <= m; i++)
{
//cin>>z>>x>>y;
scanf("%d%d%d", &z,&x,&y);
if(z == 1)
{
int u = find(x);
int v= find(y);
f[u] = v;
}
else if(z == 2)
{
int u = find(x);
int v = find(y);
if( u == v)
printf("%c\n", 'Y');
else
printf("%c\n", 'N');
}
}
//system("pause");
}