HDU-1083 Courses

简单的二分图匹配

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int N=500+10;
const int M=6e4+10;
struct Edge
{
    int to,nxt;
}edge[M];
int tot,first[N];
void addedge(int u,int v)
{
    edge[tot].to=v;
    edge[tot].nxt=first[u];
    first[u]=tot++;
}
void init()
{
    tot=0;
    memset(first,-1,sizeof(first));
}
int linker[N];
bool used[N];
bool dfs(int u)
{
    for(int i=first[u];i!=-1;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(!used[v])
        {
            used[v]=true;
            if(linker[v]==-1||dfs(linker[v]))
            {
                linker[v]=u;
                return true;
            }
        }
    }
    return false;
}
int uN;
int hungary()
{
    int res=0;
    memset(linker,-1,sizeof(linker));
    for(int i=1;i<=uN;i++)
    {
        memset(used,false,sizeof(used));
        if(dfs(i)) res++;
    }
    return res;
}
int main()
{
    int T;
    int p,n,cnt,a;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d%d",&p,&n);
        for(int i=1;i<=p;i++)
        {
            scanf("%d",&cnt);
            for(int j=0;j<cnt;j++)
            {
                scanf("%d",&a);
                addedge(i,a+p);
            }
        }
        uN=p;
        if(hungary()==p) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}