POJ-1469 COURSES

题目链接：http://poj.org/problem?id=1469

题目大意：

给你p门课程和n个学生，一个学生可以选0门，1门，或者多门课程，现在要求一个由p个学生组成的集合，满足下列2个条件：

1.每个学生选择一个不同的课程

2.每个课程都有不同的代表

如果满足，就输出YES

解题思路：

二分图最大匹配。

邻接矩阵map[i][j]表示j号学生喜欢i号课程，然后对课程进行寻找匹配，匹配成功则标记学生，之后统计匹配数目即可。

代码如下：

#include
#include
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;
}