Divide Groups（分组）（二分图染色）

题目链接

题目大意是说输入数字n

然后告诉你第i个人都认识谁？

让你把这些人分成两堆，使这每个堆里的人都互相认识。

做法：把不是互相认识的人建立一条边，则构建二分图，两堆的人肯定都互相认识，也就是说，互相认识的两个人肯定不相连。

——代码

#include <cstdio>
#include <cstring>

using namespace std;

int n, cnt;
int head[200001], to[200001], next[200001], color[101];
bool know[101][101];

void add(int x, int y)
{
    to[cnt] = y;
    next[cnt] = head[x];
    head[x] = cnt++;
}
//把点染成1或-1 
bool dfs(int u, int c)
{
    int i, v;
    color[u] = c;
    for(i = head[u]; i != - 1; i = next[i])
    {
        v = to[i];
        if(color[v] == c) return 0;
        if(color[v] == 0 && !dfs(v, -c)) return 0;
    }
    return 1;
}

bool solve()
{
    int i;
    for(i = 1; i <= n; i++)
     if(color[i] == 0 && !dfs(i, 1))
      return 0;
    return 1;
}

int main()
{
    int i, j, x;
    while(~scanf("%d", &n))
    {
        memset(know, 0, sizeof(know));
        memset(color, 0, sizeof(color));
        for(i = 1; i <= n; i++)
          while(scanf("%d", &x) && x)
          know[i][x] = 1;
        memset(head, -1, sizeof(head));
        for(i = 1; i <= n; i++)
         for(j = i + 1; j <= n; j++)
          if(!know[i][j] || !know[j][i])
          {
              add(i, j);
              add(j, i);
          }
        if(solve()) printf("YES
");
        else printf("NO
");
    }
    return 0;
}