比较裸的二分图匹配,意思是说每个学生有1门或多门课程,现在要求每门课程有一个学生能代表,并且每个学生只能代表一门课程,问你是否可以选出一些学生代表所有的课程。
用网络流做的二分图最大匹配。
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int inf=1<<29;
const int maxn=500;
const int maxm=80000;
int e,st,des,p,n,m,head[maxn],pnt[maxm],nxt[maxm],flow[maxm],level[maxn],q[maxn];
void AddEdge(int u,int v,int f)
{
pnt[e]=v;nxt[e]=head[u];flow[e]=f;head[u]=e++;
pnt[e]=u;nxt[e]=head[v];flow[e]=0;head[v]=e++;
}
bool BFS()
{
memset(level,0,sizeof(level));
int pre=0,last=1;
level[st]=1;
q[0]=st;
while(pre<last)
{
int u=q[pre++];
if(u==des)
return true;
for(int i=head[u];i!=-1;i=nxt[i])
if(flow[i]&&!level[pnt[i]])
{
level[pnt[i]]=level[u]+1;
q[last++]=pnt[i];
}
}
return false;
}
int DFS(int u,int sum)
{
if(u==des)
return sum;
for(int i=head[u],t;i!=-1;i=nxt[i])
if(flow[i]&&level[pnt[i]]==level[u]+1&&(t=DFS(pnt[i],min(sum,flow[i]))))
{
flow[i]-=t;
flow[i^1]+=t;
return t;
}
return level[u]=0;
}
void maxflow()
{
int ans=0;
while(BFS())
while(1)
{
int val=DFS(st,inf);
if(!val)
break;
ans+=val;
}
printf("%s\n",ans==p?"YES":"NO");
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
e=st=0;des=p+n+1;
memset(head,-1,sizeof(head));
scanf("%d%d",&p,&n);
for(int i=1;i<=p;i++)
{
scanf("%d",&m);
while(m--)
{
int v;
scanf("%d",&v);
AddEdge(i,v+p,1);
}
}
for(int i=1;i<=p;i++)
AddEdge(st,i,1);
for(int i=1;i<=n;i++)
AddEdge(p+i,des,1);
maxflow();
}
return 0;
}