hdu4751(二分图判定)

链接：点击打开链接

题意：给出n个人的关系，现在问是否可以分成两个集合使得每个集合中的人都互相认识

代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int n,sig,col[105],s[105][105];
void dfs(int S,int co){
    int i;
    if(sig)
    return;
    col[S]=co;
    for(i=1;i<=n;i++){
        if(s[S][i]==0){
            if(col[i]==-1)
            dfs(i,co^1);
            else{
                if(col[i]==col[S]){
                    sig=1;
                    return;
                }
            }
        }
    }
}
int main(){
    int i,j,u;
    while(scanf("%d",&n)!=EOF) {
        memset(s,0,sizeof(s));
        memset(col,-1,sizeof(col));
        for(i=1;i<=n;i++)
        s[i][i]=1;
        for(i=1;i<=n;i++){
            while(scanf("%d",&u)&&u)
            s[i][u]=1;
        }
        for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
        if(s[i][j]==0||s[j][i]==0)
        s[i][j]=s[j][i]=0;                      //只处理两两相互不认识的，两两都认识的
        sig=0;                                  //怎么染都可以，就是一定保证不认识的在两个集合
        for(i=1;i<=n;i++){
            if(col[i]==-1){                     //图可能不连通，每个强连通分量都要dfs
                dfs(i,0);
                if(sig)
                break;
            }
        }
        if(sig)
        puts("NO");
        else
        puts("YES");
    }
    return 0;
}