二分图匹配学习笔记(网络流)

之前学二分图写的匈牙利qwq,现在学完网络流回去写。。觉得比匈牙利好理解多了啊qwq。
建一个虚拟源点和虚拟汇点,连边权为1的边。然后算最大流。
以下代码为输出方案版:

#include
using namespace std;

const int MAXN=1e5+5;
const int INF=1e7;

struct edge{
    int to,next,w,u;
}e[MAXN<<1];

int num=0;
int head[MAXN],cur[MAXN],pre[MAXN],cnt=1;

int fangan[105][105];
inline void add(int u,int v,int w){e[++cnt]=(edge){v,head[u],w,u},head[u]=cnt;}

int m,n;

queue<int>q;
int dep[MAXN];
bool bfs(int x){
    memset(dep,0,sizeof(dep));
    q.push(x);
    dep[x]=1;
    while(q.size()){
        int u=q.front();q.pop();
        for(int i=head[u];i;i=e[i].next){
            int v=e[i].to,w=e[i].w;
            if(w>0&&!dep[v]){
                dep[v]=dep[u]+1;q.push(v);
            }
        }   
    }
    if(!dep[n+1])return 0;
    return 1;
}

int dfs(int u,int flow){
    if(u==n+1)return flow;
    for(int &i=cur[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(w>0&&dep[v]==dep[u]+1){
            int tem=dfs(v,min(flow,w));
            if(tem>0){
                e[i].w-=tem;
                e[i^1].w+=tem;
               // if(e[i].u)cout<"<
                return tem;
            }
        }
    }
    return 0;
}

int vis[MAXN];

int dinic(){
    int ans=0;
    while(bfs(0)){
        for(int i=0;i<=n+1;i++)cur[i]=head[i];
        while(int d=dfs(0,INF)){
            ans+=d; 
        }
    }
    return ans;
}

int main(){
    memset(fangan,0,sizeof(fangan));
    scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++){
        add(0,i,1);
        add(i,0,0);
    }
    for(int i=m+1;i<=n;i++){
        add(i,n+1,1);
        add(n+1,i,0);
    }
    int a=0,b=0;
    while(1){
        scanf("%d%d",&a,&b);
        if(a==-1&&b==-1)break;
        add(a,b,1);
        add(b,a,0);
    }
    int ans=dinic();
    if(ans){
        printf("%d\n",ans);
        for(int i=1;i<=m;i++)
            for(int j=head[i];j;j=e[j].next){
                int v=e[j].to,w=e[j].w;
                if(!w&&v>m)printf("%d %d\n",i,v);
            }
    }
    else printf("No Solution!");
    return 0;
}

你可能感兴趣的:(二分图,网络流)