HDU 4751 Divide Groups(二分图判断)

给出的图是一张有向图，代表i认识j，那么line[i][j]==false||line[j][i]==false代表i和j不是相互认识，那么他们肯定不可以在一个集合里，所以i和j之间连边，最后判断是不是二分图，如果是，那么可以分，否则不能分。

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=100+5;
int head[maxn*maxn],to[maxn*maxn],Next[maxn*maxn];
bool line[maxn][maxn];
int color[maxn];
int n;
int tot;
void addedge(int u,int v)
{
    to[tot]=v;
    Next[tot]=head[u];
    head[u]=tot++;
}
bool getc(int u)
{
    for(int i=head[u];i!=-1;i=Next[i])
    {
        int v=to[i];
        if(color[v]==-1){
            color[v]=3-color[u];
            if(!getc(v)) return false;
        }
        else if(color[u]==color[v]) return false;
    }
    return true;
}
int main()
{
    while(~scanf("%d",&n))
    {
        tot=0;
        memset(line,false,sizeof(line));
        memset(head,-1,sizeof(head));
        for(int i=1;i<=n;i++){
            int v;
            while(scanf("%d",&v)&&v){
                line[i][v]=true;
            }
        }
        for(int i=1;i<=n;i++){
            for(int j=i+1;j<=n;j++){
                if(line[i][j]==false||line[j][i]==false) {addedge(i,j);addedge(j,i);}
            }
        }
        bool ok=true;
        memset(color,-1,sizeof(color));
        for(int i=1;i<=n;i++){
            if(color[i]==-1){
                color[i]=1;
                ok=getc(i);
                if(ok==false) break;
            }
        }
        if(ok) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}