二分图匹配----匈牙利算法之二

poj 1469 COURSES

//题意:现在有p门课程和n个学生,现在需要有由p个学生组成的组织,该组织满足每个学生代表一门课程以及每门课程只能由一个学生代表,

//现给出每门课程都有哪些学生可以参加,要求判断是否存在满足题意所需要的组织.

//求解二分图的最大匹配数,如果与p相等就输出"YES",否则输出"NO".

#include<iostream>        //二分图的最大匹配
#include<cstring>
using namespace std;
int p,n;
int result[301];    //result记录V2中的点匹配的点的编号
int edge[101][301],vis[301];        //vis记录V2中的每个点是否被搜索过
bool find(int a)
{
    for(int i=1;i<=n;++i)
    {
        if(edge[a][i]==1&&vis[i]==0)        //如果节点i与a相邻并且未被查找过
        {
            vis[i]=1;        //标记i为已查找过
            if( result[i]==0 || find(result[i])==true )        //如果i未在前一个匹配M中或者 i在匹配M中，但是从与i相邻的节点出发可以有增广路
            {
                result[i]=a;        //记录查找成功记录
                return true;        //返回查找成功
            }
        }
    }
    return false;
}
int main()
{
    int cases;
    scanf("%d",&cases);
    while(cases--)
    {
        memset(edge,0,sizeof(edge));
        memset(result,0,sizeof(result));
        scanf("%d%d",&p,&n);
        int m,t;
        for(int i=1;i<=p;++i)
        {
            scanf("%d",&m);
            while(m--)
            {
                scanf("%d",&t);
                edge[i][t]=1;
            }
        }
        int ans=0;
        for(int i=1;i<=p;++i)
        {
            memset(vis,0,sizeof(vis));        //清空上次搜索时的标记
            if(find(i)==true)        //从节点i尝试扩展
                ans++;    
        }
        if(ans==p)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}