原题见UOJ78
有n位男生,m位女生,其中有一些男生和女生愿意结成配偶。
请问这个班级里最多产生多少对配偶?

匈牙利算法是一种用增广路求二分图最大匹配的算法。

算法流程:
给每一个男生尝试寻找妹子:
对于每一个他愿意结成配偶的妹子,如果这个妹子没有和别人结成配偶,那么他们结成配偶。
如果这个妹子和别人结成了配偶,如果这个妹子的配偶还能找到另一个妹子,那么这个妹子和这个男生结成配偶,这个妹子的原配和他找到的妹子结成配偶。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define up(a,b,c) for (int a(b),end##a(c);a<=end##a;++a)
#define down(a,c,b) for (int a(b),end##a(c);a>=end##a;--a)
#define uup(a,c,b) for (int a(b),end##a(c);a<=end##a;++a)
#define udown(a,b,c) for (int a(b),end##a(c);a>=end##a;--a)
using namespace std;
int read(){
    char s;
    int k=0,base=1;
    while((s=getchar())!='-'&&s!=EOF&&!(s>='0'&&s<='9'));
    if(s==EOF)exit(0);
    if(s=='-')base=-1,s=getchar();
    while(s>='0'&&s<='9'){
        k=k*10+(s-'0');
        s=getchar();
    }
    return k*base;
}
void write(int x){
    if(x<0){
        putchar('-');
        write(-x);
    }
    else{
        if(x/10)write(x/10);
        putchar(x%10+'0');
    }
}
int a[1024][1024],pip[1024],f_pip[1024],vis[1024];
int n,m,e,tot;
int dfs(int x){//尝试为男生x找妹子
    up(i,1,m)if(a[x][i]&&!vis[i]){//如果男生x和女生y可以结成配偶
        vis[i]=1;//vis数组是为了防止环形寻找
        if(!f_pip[i]||dfs(f_pip[i])){//如果女生y没有配偶或者给女生y的配偶能找到配偶
            pip[x]=i;//男生x找到了女生i
            f_pip[i]=x;//女生i找到了男生x
            return 1;//成功
        }
    }
    return 0;
}
int main(){
    n=read();
    m=read();
    e=read();
    up(i,1,e){
        int x=read(),y=read();
        if(y>m||x>n||n<1||y<1)continue;
        a[x][y]=1;
    }
    up(i,1,n){
        up(j,1,m)vis[j]=0;
        tot+=dfs(i);
    }
    printf("%d\n",tot);
    up(i,1,n){
        printf("%d%c",pip[i],i==n?'\n':' ');
    }
    return 0;
}

标签: 模板,

添加新评论