hdu 3605 Escape（二分图多重匹配问题）

/*
题意：末日逃亡，n个人逃往m个星球，一个人只能在某些星球上生存。一个星球最多居住人口有上限。求最后是否可以让所有人都逃离地球。

题解：
基础的二分图多重匹配问题（类似于二分图最大匹配）

技巧：
①处，可以减少代码量
*/


#include <iostream>
#define re(i, n) for(int i = 0; i < n; ++ i)//①
using namespace std;
const int nMax = 100007;
const int mMax = 12;
int n, m;
int map[nMax][mMax];//map[i][j] = 1表示：i适合在j星球上生存
int cap[mMax];//星球上人口上限
int num[mMax];//统计星球上人口的数量
int link[mMax][nMax];//link[i][j] = k表示：在i星球上居住的第j个人是k
int useif[mMax];

int dfs(int k)
{
	re(i, m)
	{
		if(!useif[i] && map[k][i])
		{
			useif[i] = 1;
			if(num[i] < cap[i])//人口数小于星球上限
			{
				link[i][num[i] ++] = k;
				return 1;
			}
			re(j, cap[i])
			{
				if(dfs(link[i][j]))//link[i][j]可以找到别的匹配星球
				{
					link[i][j] = k;
					return 1;
				}
			}
		}
	}
	return 0;
}

int maxMatch()
{
	memset(num, 0, sizeof(num));
	re(i, n)//查看n个人是否都可以找到适合的星球
	{
		memset(useif, 0, sizeof(useif));
		if(!dfs(i)) return 0;
	}
	return 1;
}

int main()
{
	//freopen("f://data.in", "r", stdin);
	while(scanf("%d%d", &n, &m) != EOF)
	{
		re(i, n) re(j, m) scanf("%d", &map[i][j]);
		re(i, m) scanf("%d", &cap[i]);
		printf("%s\n", maxMatch() ? "YES" : "NO");
	}
	return 0;
}