POJ 珍惜昆虫的繁殖问题

#include <iostream>
#include <cstdio>
#include <cstring>
#define MAX_N 5000

#define MAX_K 1000005
using namespace std;
//完成n个元素的初始化
int x[MAX_K];
int y[MAX_K];
int n,k;
int par[MAX_N];
int rank[MAX_N];
void init(int n)
{
for(int i=0;i<n;i++)
{
par[i]=i;
rank[i]=0;
}
}

//查询树的根
int find(int x)
{
if(par[x]==x)
return x;
else
return par[x]=find(par[x]);//路径压缩，对于每个节点，一旦向上走到了一次根节点，就把这个店到父亲的边改为直接连向边

}

//合并x和y所属的集合
void unite(int x,int y)
{
//找到x和y的根，并判断根是否相同
x=find(x);
y=find(y);
if(x==y)
return ;
if(rank[x]<rank[y])
{
//将x设为y的子树
par[x]=y;
}
else{
par[y]=x;
if(rank[x]==rank[y])
rank[x]++;
}
}
bool same(int x,int y)
{
//只需要判断他们的根节点是否相同即可
return (find(x)==find(y));
}

int t,g;
int main()
{
scanf("%d",&t);
g=t;
while(t--)
{
scanf("%d %d",&n,&k);
for(int i=0;i<k;i++)
{
scanf("%d %d",x+i,y+i);
}
void solve();
solve();
}

}
void solve()
{
int ans=0;
init(n*2);
for(int i=0;i<k;i++)
{
int a=x[i]-1;
int b=y[i]-1;
if(a<0||b<0||a>=n||b>=n)
{
printf("Scenario #%d:\nSuspicious bugs found!\n\n",g-t);
return ;
}
if(same(a,b))
{
printf("Scenario #%d:\nSuspicious bugs found!\n\n",g-t);
return ;
}
else{
unite(a,b+n);
unite(a+n,b);
}
}
printf("Scenario #%d:\nNo suspicious bugs found!\n\n",g-t);

}


抛开题意有多open不说，我们需要完成的就是对昆虫分类，分为男和女，由于n的范围是小于等于2000，我们可以对比食物链的那道题，将par[5]；

分为两类：a，b是同一性别

A+n和b，b+n和a是异性