洛谷—— 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%的数据，n\le 100,m\le 20000n≤100,m≤20000

 #include <cstring>
 #include <cstdio>

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

 const int M();
 const int N();

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

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

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

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

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

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