上次也写过这样的题,这次时隔三秋我们再做这道题已经有不一样的感受了。
题目没变,还是那个把陌生人放在不同的地方的问题,这次我用并查集去优化了一下它,因为在同一棵树上的节点不会受到其他树上节点的影响,所以我把这样的点放在一颗树上即可,那么,我们就不用去遍历每一个节点找匹配了,只需要去看树即可了,代码风格还是连接上的没变(变量名也没变,可以先看上一章在过来理解这一章),这次熟悉了很多,一气呵成就没标注释了。
#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;
}