一道匈牙利算法的水题,题目让求恰好满足:1,每个学生都要选一门课程;2,每个课程都有一个学生选择。
很明显,求出二分图的最大匹配数,如果等于课程数目,就YES,不等就NO;
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define N 310
int n,p;
int used[N],mat[N][N],link[N];
int dfs(int t)
{
int i;
for(i=1; i<=p; i++)
{
if(used[i]==0&&mat[t][i])
{
used[i]=1;
if(link[i]==-1||dfs(link[i]))
{
link[i]=t;
return 1;
}
}
}
return 0;
}
int Find()
{
int num=0;
int i;
for(i=1; i<=n; i++)
{
memset(used,0,sizeof(used));
if(dfs(i))
num++;
}
if(num!=p)
return 0;
return 1;
}
int main()
{
int t,i,k,a;
scanf("%d",&t);
while(t--)
{
memset(mat,0,sizeof(mat));
memset(link,-1,sizeof(link));
scanf("%d%d",&p,&n);
for(i=1; i<=p; i++)
{
scanf("%d",&k);
while(k--)
{
scanf("%d",&a);
mat[a][i]=1;
}
}
if(Find())
printf("YES\n");
else
printf("NO\n");
}
return 0;
}
//********实在想不明白用G++和用C++提交有什么区别,当时用C++提交,超时了N遍,换成G++,果断AC。。。。。不明白C++为何会超时!