hdu3605(二分图多重匹配伪模板)

题目描述

给出\(n\)个人,\(m\)个星球,每个星球有一个移民人数限制\(lim\)，现在询问你一个最好的匹配方式,使移民人数最多。

\(n\) (1 \(\le\) \(n\) \(\le\) 100000), \(m\) (1 \(\le\) \(m\) \(\le\) 10)

虽然说是多重匹配,但这里只有星球是1\(\le\)\(lim\)且\(m\)比较小

\(m\)和\(n\)都很大或都有限制时还是会用最大流

样例输入

1 1
1
1

2 2
1 0
1 0
1 1

样例输出

YES
NO

Code

#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100002;
int n,m,match[12][N],sz[N],mp[N][12],a[12];
bool vis[12];
int dfs(int s)
{
    for(int i=0;i<m;i++)
    {
        if(mp[s][i] && !vis[i])
        {
            vis[i]=true;
            if(sz[i]<a[i])
            {
                match[i][sz[i]++]=s;
                return 1;
            }
            for(int j=0;j<sz[i];j++)
            {
                if(dfs(match[i][j]))
                {
                    match[i][j]=s;
                    return 1;
                }
            }
        }
    }
    return 0;
}
int main(){
//    freopen("t.in","r",stdin);
    while(scanf("%d %d",&n,&m)==2){
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
           	 scanf("%d",&mp[i][j]);
        for(int i=0;i<m;i++)
            scanf("%d",&a[i]);
        memset(sz,0,sizeof(sz));
        bool flag=true;
        for(int i=0;i<n;i++)
        {
            memset(vis,false,sizeof(vis));
            if(!dfs(i))
            {
                flag=false;
                break;
            }
        }
        if(flag) puts("YES");
        else puts("NO");
//        for(int i=0;i<m;++i)printf("%d ",sz[i]); puts("");
    }
    return 0;    
}