E - Courses
Consider a group of N students and P courses. Each student visits zero, one or more than one courses. Your task is to determine whether it is possible to form a committee of exactly P students that satisfies simultaneously the conditions:
. every student in the committee represents a different course (a student can represent a course if he/she visits that course)
. each course has a representative in the committee
Your program should read sets of data from a text file. The first line of the input file contains the number of the data sets. Each data set is presented in the following format:
P N
Count1 Student1 1 Student1 2 ... Student1 Count1
Count2 Student2 1 Student2 2 ... Student2 Count2
......
CountP StudentP 1 StudentP 2 ... StudentP CountP
The first line in each data set contains two positive integers separated by one blank: P (1 <= P <= 100) - the number of courses and N (1 <= N <= 300) - the number of students. The next P lines describe in sequence of the courses . from course 1 to course P, each line describing a course. The description of course i is a line that starts with an integer Count i (0 <= Count i <= N) representing the number of students visiting course i. Next, after a blank, you'll find the Count i students, visiting the course, each two consecutive separated by one blank. Students are numbered with the positive integers from 1 to N.
There are no blank lines between consecutive sets of data. Input data are correct.
The result of the program is on the standard output. For each input data set the program prints on a single line "YES" if it is possible to form a committee and "NO" otherwise. There should not be any leading blanks at the start of the line.
An example of program input and output:
Input
2
3 3
3 1 2 3
2 1 2
1 1
3 3
2 1 3
2 1 3
1 1
Output
YES
NO
要求每班有一名同学作为代表出来组成委员会,每位代表只代表一门课程,并且他必须参与这门课程。
其实就是将课程看成集合1,学生看成集合2,求这两个点集的最大匹配。使用匈牙利算法即可直接得到最大匹配。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
#include <queue>
using namespace std;
const int maxn = 304;
int P, N, T;
int pre[maxn];
bool g[maxn][maxn], vis[maxn];
bool dfs(int u){
for(int i = 1; i <= N; i++){
if(g[u][i] && !vis[i]){
vis[i] = true;
if(pre[i] == -1 || dfs(pre[i])){
pre[i] = u;
return true;
}
}
}
return false;
}
int check(){
memset(pre, -1, sizeof(pre));
int ret = 0;
for(int i = 1; i <= P; i++){
memset(vis, false, sizeof(vis));
if(dfs(i))
ret++;
}
return ret;
}
int main(){
#ifdef TEST
freopen("test.txt", "r", stdin);
#endif // TEST
cin >> T;
while(T--){
memset(g, false, sizeof(g));
cin >> P >> N;
for(int i = 1; i <= P; i++){
int k, q;
scanf("%d", &k);
for(int j = 1; j <= k; j++){
scanf("%d", &q);
g[i][q] = true;
}
}
int res = check();
if(res == P)
cout << "YES" << endl;
else
cout << "NO" << endl;
}
return 0;
}