洛谷—— P2417 课程

https://www.luogu.org/problemnew/show/2417

题目描述

n个学生去p个课堂，每一个学生都有自己的课堂，并且每个学生只能去一个课堂，题目要求能够安排每一个课堂都有人吗？

输入输出格式

输入格式：

第一行是测试数据的个数，

每组测试数据的开始分别是p和n，

接着p行，每行的开始是这个课堂的学生人数m，接着m个数代表该课堂的学生编号

输出格式：

如果该组数据能够这样安排就输出YES，否则输出NO。

输入输出样例

输入样例#1： 复制

2
3 3
3 1 2 3
2 1 2
1 1
3 3
2 1 3
2 1 3
1 1

输出样例#1： 复制

YES
NO

说明

对于100%的数据，nle 100,mle 20000n≤100,m≤20000

#include <cstring>
#include <cstdio>

inline void read(int &x)
{
    x=0; register char ch=getchar();
    for(; ch>'9'||ch<'0'; ) ch=getchar();
    for(; ch>='0'&&ch<='9'; ch=getchar()) x=x*10+ch-'0';
}

const int M(20005);
const int N(110);

int n,p,sumvis;
bool link[N][M];
int vis[M],mat[M];

bool find(int u)
{
    for(int v=1; v<=n; ++v)
        if(vis[v]!=sumvis&&link[u][v])
        {
            vis[v]=sumvis;
            if(!mat[v]||find(mat[v]))
            {
                mat[v]=u;
                return 1;
            }
        }
    return false;
}

inline bool work()
{
    int m,cnt=0;
    read(p),read(n);
    for(int u=1; u<=p; ++u)
    {
        read(m);
        for(int v; m--; )
          read(v),link[u][v]=1;
    }
    for(int i=1; i<=p; ++i)
    {
        sumvis++;
        cnt+=find(i);
    }
    return cnt==p;
}

inline void init()
{
    sumvis=0;
    memset(mat,0,sizeof(mat));
    memset(vis,0,sizeof(vis));
    memset(link,0,sizeof(link));
}

int Presist()
{
    int t; read(t);
    for(; t--; init())
      if(!work()) puts("NO");
      else puts("YES");
    return 0;
}

int Aptal=Presist();
int main(int argc,char**argv){;}