13-14寒假作业9

poj2524 并查集水题

输入点的总数，在输入几对点，表示这两个点在同一组里，问总共有多少组。

并查集是很快判断这种题的一个办法，用到了一些树的思想。首先设一个数组f[i]，表示第i个点的祖宗是谁，开始的时候每个人的祖宗都是自己。一个祖宗和它的后代们就是在同一个集合里的所有点。

每输入两个个数字，我们都找到他们的祖宗是谁，如果是同一个祖宗，说明他们是同一个集合的，否则令一个数的祖宗是另一个数。

int find(int x)

{

return x==f[x]?x:f[x]=find(f[x]);

}

这个函数的返回值是点x的祖宗。最坏的时候，可能这棵树是一个链表，上边的函数就避免了一个问题，递归找到最后的祖宗，返回的时候把所有的点都直接变成他的儿子。这个函数让我获益不少。

#include<cstdio>

#include<cstring>

const int maxn=1000;

int x,y,OK,temp;

int f[maxn];

bool u[maxn];

int find(int x)

{

return x==f[x]?x:f[x]=find(f[x]);

}

int main()

{

int t=1,i,a1,a2;

temp=1;

while(scanf("%d%d",&x,&y),x!=-1&&y!=-1)

{

if(temp&&!x&&!y)

{

printf("Case %d is a tree.\n",t++);

continue;

}

if(temp)

{

a1=a2=0;

OK=1;

memset(u,1,sizeof(u));

for(i=0;i<maxn;++i)

f[i]=i;

}

a2++;

if(temp) temp=0;

if(!x&&!y)

{

temp=1;

if(a1!=a2) OK=0;

OK?printf("Case %d is a tree.\n",t++):printf("Case %d is not a tree.\n",t++);

continue;

}

else

{

if(f[y]=find(x)==find(y)) OK=0;

if(u[x]) {a1++; u[x]=0;}

if(u[y]) {a1++; u[y]=0;}

}

}

return 0;

}