题目链接:http://poj.org/problem?id=1469
题目大意:
给你p门课程和n个学生,一个学生可以选0门,1门,或者多门课程,现在要求一个由p个学生组成的集合,满足下列2个条件:
1.每个学生选择一个不同的课程
2.每个课程都有不同的代表
如果满足,就输出YES
解题思路:
二分图最大匹配。
邻接矩阵map[i][j]表示j号学生喜欢i号课程,然后对课程进行寻找匹配,匹配成功则标记学生,之后统计匹配数目即可。
代码如下:
#include<cstdio>
#include<cstring>
using namespace std;
#define P 110
#define N 310
int map[P][N];
int match[N];
bool use[N];
int p, n;
bool find(int u) //u是课程
{
for(int i = 1; i <= n; ++i)
{
if(!use[i] && map[u][i]) //标记匹配的学生
{
use[i] = true;
if(match[i] == - 1 || find(match[i]))
{
match[i] = u;
return true;
}
}
}
return false;
}
int sum()
{
int sumall = 0;
for(int i = 1; i <= p; ++i) //统计匹配的课程
{
memset(use, false, sizeof(use));
if(find(i))
sumall++;
}
return sumall;
}
int main()
{
int ncase;
int stunum, temp, ans;
scanf("%d", &ncase);
while(ncase--)
{
memset(map, 0, sizeof(map));
memset(match, -1, sizeof(match));
scanf("%d%d", &p, &n);
for(int i = 1; i <= p; ++i)
{
scanf("%d", &stunum);
for(int j = 1; j <= stunum; ++j)
{
scanf("%d", &temp);
map[i][temp] = 1; //j号学生喜欢i号课程
}
}
ans = sum();
printf("%s\n", ans == p ? "YES" : "NO");
}
return 0;
}