HDU 4751 Divide Groups（判断二分图）

#include<map>
#include<queue>
#include<cmath>
#include<iostream>
#include<cstdio>
#include<stack>
#include<cstring>
#include<algorithm>
#define LL long long
#define inf 0x3f3f3f3f
const double PI=acos(-1.0);
using namespace std;
struct node{
int to,next;
}q[50000];
int cnt,mp[500][500],head[50000],cro[50000],col[50000];
bool vis[50000];
void bu(int a,int b){
q[cnt].to=b;
q[cnt].next=head[a];
head[a]=cnt++;
}
bool color(int x){
for(int i=head[x];~i;i=q[i].next){
int u=q[i].to;
if(!col[u]){
col[u]=!col[x];
if(!color(u))
return false;
}
else if(col[x]==col[u]){
return false;
}
}
return true;
}
int main(){
int n,m,i,j,k,a,b,ans;
while(~scanf("%d",&n)){
cnt=ans=0;
memset(mp,0,sizeof(mp));
memset(head,-1,sizeof(head));
for(i=1;i<=n;i++){
while(~scanf("%d",&k)&&k){
//bu(i,k);
mp[i][k]=1;
}
}
for(i=1;i<=n;i++){
for(j=1;j<=n;j++){
if(i==j) continue;///注意自己不能和自己连边建图
if((!mp[i][j])){
bu(i,j);bu(j,i);
}
}
}

memset(col,0,sizeof(col));
bool p;
for(i=1;i<=n;i++){///因为不能保证图的连通性所以每个点都要进行遍历
if(col[i]==0){
p=color(i);
if(!p) break;
}
}
if(!p)
puts("NO");
else
puts("YES");
}
return 0;
}