poj1703 并查集

时间:2021-12-29 06:48:39

输入是2个不在一起的人,可以用一个数组来保存和他矛盾的人。这样find的时候就find(hash[]);就可以;

#include<stdio.h>
#include<string.h>
int pa[],h[],n;
void init()
{
for(int i=;i<=n;i++)
{
pa[i]=i;
h[i]=;
}
}
int find(int x)
{
if(x!=pa[x])
pa[x]=find(pa[x]);
return pa[x];
}
int main()
{
int i,j,t,m;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
getchar();
init();
char s;
int a,b;
for(i=;i<m;i++)
{
scanf("%c %d %d",&s,&a,&b);
getchar();
if(s=='D')
{
if(!h[a])
h[a]=b;
else
{
int l1,l2;
l1=find(h[a]);
l2=find(b);
pa[l1]=l2;
}
if(!h[b])
h[b]=a;
else
{
int l1,l2;
l1=find(h[b]);
l2=find(a);
pa[l1]=l2;
} }
else
{
/*if(n==2)
{
if(a!=b)
printf("In different gangs.\n");
else
printf("In the same gang.\n");
continue;
}*/
int l1=find(a);
int l2=find(b);
int l3=find(h[b]);
if(h[l1]==||h[l2]==)
{
printf("Not sure yet.\n");
continue;
}
if(l1==l2)
printf("In the same gang.\n");
else if(l1==l3)
{
printf("In different gangs.\n");
}
else
printf("Not sure yet.\n"); }
}
}
}