Divide Groups 【HDU - 4751】【DFS+拓扑+并查集优化】

上次也写过这样的题，这次时隔三秋我们再做这道题已经有不一样的感受了。

  题目没变，还是那个把陌生人放在不同的地方的问题，这次我用并查集去优化了一下它，因为在同一棵树上的节点不会受到其他树上节点的影响，所以我把这样的点放在一颗树上即可，那么，我们就不用去遍历每一个节点找匹配了，只需要去看树即可了，代码风格还是连接上的没变（变量名也没变，可以先看上一章在过来理解这一章），这次熟悉了很多，一气呵成就没标注释了。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef long long ll;
int N;
set<int> s;
vector<int> vt[105];
queue<int> Q;
bool mp[105][105];
bool flag=true;
int root[105];
int vis[105];
int fid(int x)
{
    if(x==root[x]) return x;
    return root[x]=fid(root[x]);
}
void mix(int x, int y)
{
    int u=fid(x);
    int v=fid(y);
    if(u!=v)
    {
        root[u]=v;
    }
}
void dfs(int ss)
{
    while(!Q.empty()) Q.pop();
    vis[ss]=1;
    Q.push(ss);
    while(!Q.empty())
    {
        int u=Q.front();
        Q.pop();
        int len=(int)vt[u].size();
        for(int i=0; i<len; i++)
        {
            int v=vt[u][i];
            if(!vis[v])
            {
                vis[v]=3-vis[u];
                Q.push(v);
            }
            else
            {
                if(vis[v]==vis[u])
                {
                    flag=false;
                    return;
                }
            }
        }
    }
}
int main()
{
    while(scanf("%d",&N)!=EOF)
    {
        memset(mp, false, sizeof(mp));      //初始化
        memset(vis, 0, sizeof(vis));
        s.clear();
        flag=true;
        for(int i=0; i<=N; i++)
        {
            root[i]=i;
            vt[i].clear();
        }
        for(int i=1; i<=N; i++)
        {
            int e1;
            while(scanf("%d",&e1)&&e1)  mp[i][e1]=true;
        }
        for(int i=1; i<=N; i++)
        {
            for(int j=1; j<i; j++)
            {
                if(!mp[i][j] || !mp[j][i])
                {
                    vt[i].push_back(j);
                    vt[j].push_back(i);
                    mix(i, j);
                }
            }
        }
        for(int i=1; i<=N; i++) s.insert(fid(i));
        while(s.size() && flag)
        {
            dfs(*s.begin());
            s.erase(s.begin()++);
        }
        if(flag) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}