特殊三分图匹配

特殊三分图匹配

@LOJ

一般三分图的匹配需要运用基于拉格朗日松弛的分支定界法，并运用启发式算法得到较优的初始下界。已被证明是NPC问题。出题者在此说明一般三分图的匹配可以解决本题。
【题目描述】
三个点集X，Y，Z，同一点集之间没有边，XZ点集之间没有边。
XY点集之间有边，YZ点集之间有边。
求三分图的最大匹配。三分图匹配是指有不相交的若干个点集，｛i,j,k(i∈X,j∈Y,k∈Z)｝
【输入格式】
第1行n1,n2,n3,m1,m2(1<=n1,n2,n3<=1000)(1<=m1,m2<=5000)
分别表示XYZ点集的点数，XY之间的边数，YZ之间的边数。
然后m1行，一行两个整数a,b，表示xa和yb之间有边。
然后m2行，一行两个整数a,b，表示ya和zb之间有边。
【输出格式】
一行一个整数，三分图的最大匹配数。

【样例输入1】
3 4 5 6 6
1 1
1 3
2 2
2 4
3 1
3 3
1 2
2 1
2 3
3 2
4 4
4 5

【样例输出1】
2

【样例输入2】
3 3 3 5 4
1 2
2 3
2 2
1 3
3 1
1 2
3 3
3 2
2 1

【样例输出2】
3

【注意】
输入可能存在重边。
对于100%的数据，(1<=n1,n2,n3<=1000)(1<=m1,m2<=5000)

对没错，这毒瘤题就是我出的。本题添加至LOJ6345。

鸣谢LOJdalao hack 标程

鸣谢LOJdalao 提供正解

把Y点集拆成2个点，两点之间连一条流量上限为1的流，其他边流量上限为1，添加源点和汇点，做最大流即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
const int maxn=5000;
const int oo=1000000000;

int n1,n2,n3,m1,m2;

struct Edge{
    int from,to,cap,flow;
};
vector<int>G[maxn];
vector<Edge>edges;
void Addedge(int x,int y,int z){
    Edge e;
    e.from=x;e.to=y;e.cap=z;e.flow=0;
    edges.push_back(e);
    e.from=y;e.to=x;e.cap=0;e.flow=0;
    edges.push_back(e);
    int c=edges.size();
    G[x].push_back(c-2);
    G[y].push_back(c-1);
}

int s,t;
int vis[maxn];
int d[maxn];
queue<int>q;
int Bfs(){
    memset(vis,0,sizeof(vis));
    d[s]=0;vis[s]=1;q.push(s);
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=0;i<G[x].size();++i){
            Edge e=edges[G[x][i]];
            if((!vis[e.to])&&(e.cap>e.flow)){
                d[e.to]=d[x]+1;
                vis[e.to]=1;
                q.push(e.to);
            }
        }
    }
    return vis[t];
}

int Dfs(int x,int a){
    if((x==t)||(a==0))return a;

    int nowflow=0,f=0;
    for(int i=0;i<G[x].size();++i){
        Edge e=edges[G[x][i]];
        if((d[x]+1==d[e.to])&&((f=Dfs(e.to,min(a,e.cap-e.flow)))>0)){
            nowflow+=f;a-=f;
            edges[G[x][i]].flow+=f;
            edges[G[x][i]^1].flow-=f;
            if(a==0)break;
        }
    }
    return nowflow;
}

int Maxflow(){
    int flow=0;
    while(Bfs())flow+=Dfs(s,oo);
    return flow;
}


int main(){
    scanf("%d%d%d%d%d",&n1,&n2,&n3,&m1,&m2);

    s=n1+n2+n2+n3+1;t=s+1;
    for(int i=1;i<=n1;++i)Addedge(s,i,1);
    for(int i=1;i<=n3;++i)Addedge(n1+n2+n2+i,t,1);
    for(int i=1;i<=n2;++i)Addedge(n1+i,n1+n2+i,1);
    while(m1--){
        int x,y;
        scanf("%d%d",&x,&y);
        Addedge(x,n1+y,1);
    }
    while(m2--){
        int x,y;
        scanf("%d%d",&x,&y);
        Addedge(n1+n2+x,n1+n2+n2+y,1);
    }

    printf("%d\n",Maxflow());
    return 0;
}