给出每个人适合住的星球信息和该星球能住多少人
第一行给出n m 代表有 n 个人 m 个星球
然后接下来n行每行m个数字 1代表适合第 i 个星球 0 代表不适合第 i 个星球
最后一行m个数表示第 i 个星球最多可以住多少个人
问是不是所有人都可以住到星球上
匈牙利算法可以解决多重匹配,原理和二分图最大匹配很像。注意不要把可以匹配多个的点分割然后按照正常的二分匹配来做,那样肯定会挂的。
解决多重匹配就是记录一下多重匹配的点(简称Y方点)已经匹配了Pi个点。如果Pi<Ki那么就直接上了,否则的话继续搜索Yi已经匹配的每一个点并将Yi染色。因为Yi搜一次就需要染色了,而且Y方点最多是10个,所以每次找增广路的深度最多是10,这样就很快了。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int maxn = 1e5+5;
vector<int> g[maxn];
int limit[15], match[15][maxn], v[15], n, m;
bool vis[15];
bool dfs(int x)
{
for(int i = 0; i < g[x].size(); i++)
{
int to = g[x][i];
if(!vis[to])
{
vis[to] = 1;
if(v[to] < limit[to])
{
match[to][v[to]++] = x;
return 1;
}
for(int j = 0; j < limit[to]; j++)
{
if(dfs(match[to][j]))
{
match[to][j] = x;
return 1;
}
}
}
}
return 0;
}
bool solve()
{
memset(v, 0, sizeof(v));
memset(match, 0, sizeof(match));
for(int i = 0; i < n; i++)
{
memset(vis, 0, sizeof(vis));
if(!dfs(i)) return 0;
}
return 1;
}
int main(void)
{
while(cin >> n >> m)
{
for(int i = 0; i < maxn; i++)
g[i].clear();
for(int i = 0; i < n; i++)
for(int j = 0; j < m; j++)
{
int t;
scanf("%d", &t);
if(t) g[i].push_back(j);
}
for(int i = 0; i < m; i++)
scanf("%d", &limit[i]);
puts(solve() ? "YES" : "NO");
}
return 0;
}
Sample Input
1 1
1
1
2 2
1 0
1 0
1 1
Sample Output
YES
NO