HDU 3605 Escape (二分图多重匹配模板)

题意:
给出每个人适合住的星球信息和该星球能住多少人 
第一行给出n m 代表有 n 个人 m 个星球
然后接下来n行每行m个数字 1代表适合第 i 个星球 0 代表不适合第 i 个星球
最后一行m个数表示第 i 个星球最多可以住多少个人

问是不是所有人都可以住到星球上


思路:

匈牙利算法可以解决多重匹配,原理和二分图最大匹配很像。注意不要把可以匹配多个的点分割然后按照正常的二分匹配来做,那样肯定会挂的。
解决多重匹配就是记录一下多重匹配的点(简称Y方点)已经匹配了Pi个点。如果Pi


代码:

#include
#include
#include
#include
using namespace std;
const int maxn = 1e5+5;
vector 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

你可能感兴趣的:(匹配问题)