poj 2492 A Bug's Life(关系并差集)

http://poj.org/problem?id=2492

题意：给出几组不同性别的虫子，判断是否存在同性恋

关系并差集，用r[]记录与根节点的关系，r[i]==0与根节点同性，r[i]==1与根节点异性

x以a为根节点，y以b为根节点，根据x,y异性的的关系，让b以a为根节点时，得出 r[y]+(待求值)r[b]+r[x]=1，(x,y异性，关系为1,y到b的关系，加上b到a的关系，加上a到x的关系，就是x与y的关系），则r[b]=1-r[x]-r[y],因为，r[]只有0,1两个数值，所以 r[b]=(r[x]+r[y]+1)%2;

参考blog http://www.haogongju.net/art/1579095

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#define nMAX 2005
using namespace std;
int f[nMAX],r[nMAX],n;
void make_set()
{
int i;
for(i=1;i<=n;i++)
f[i]=i,r[i]=0;
}
int find_root(int x)
{
int temp;
if(x!=f[x])
{
temp=f[x];
f[x]=find_root(f[x]);
r[x]=(r[x]+r[temp])%2;
}
return f[x];
}
void union_set(int x,int y,int rx,int ry)
{
f[ry]=rx;
r[ry]=(r[x]+r[y]+1)%2;
}
int main()
{
int CASE,ca,m,x,y;
bool fg;
scanf("%d",&CASE);
for(ca=1;ca<=CASE;ca++)
{
scanf("%d%d",&n,&m);
make_set();
fg=0;
while(m--)
{
scanf("%d%d",&x,&y);
if(fg)continue;
int rx=find_root(x);
int ry=find_root(y);
if(rx==ry)
{
if(r[x]==r[y])
{
fg=1;
continue;
}
}
else union_set(x,y,rx,ry);
}
printf("Scenario #%d:\n",ca);
if(fg)printf("Suspicious bugs found!\n\n");
else printf("No suspicious bugs found!\n\n");
}
return 0;
}