HDU 3081 Marriage Match II

二分图的最大匹配+并查集

每次匹配完之后，删除当前匹配到的边。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

const int MAXN=505;
int nx,ny;
int g[MAXN][MAXN];
int cx[MAXN],cy[MAXN];
int mk[MAXN];

int path(int u)
{
    for(int v=0;v<ny;v++)
    {
        if(g[u][v]&&!mk[v])
        {
            mk[v]=1;
            if(cy[v]==-1||path(cy[v]))
            {
                cx[u]=v;
                cy[v]=u;
                return 1;
            }
        }
    }
    return 0;
}

int MaxMatch()
{
    int res=0;
    memset(cx,-1,sizeof(cx));
    memset(cy,-1,sizeof(cy));
    for(int i=0;i<nx;i++)
    {
        if(cx[i]==-1)
        {
            memset(mk,0,sizeof(mk));
            res=res+path(i);
        }
    }
    return res;
}

int N,M,F;
int Father[MAXN];
int U[MAXN],V[MAXN];
int R[MAXN][MAXN];

int Find(int x)
{
    if(x!=Father[x]) Father[x]=Find(Father[x]);
    return Father[x];
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&N,&M,&F);
        nx=N,ny=N;
        memset(g,0,sizeof g);
        memset(R,0,sizeof R);
        for(int i=0;i<=N;i++) Father[i]=i;
        for(int i=1;i<=M;i++)
        {
            scanf("%d%d",&U[i],&V[i]);
            U[i]--;V[i]--;
            R[U[i]][V[i]]=1;
            g[U[i]][V[i]]=1;
        }
        for(int i=1;i<=F;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            u--;v--;
            u=Find(u);v=Find(v);
            if(u!=v) Father[u]=v;
        }
        for(int i=0;i<N;i++)
            for(int j=0;j<N;j++)
            {
                if(g[i][j]==1) continue;
                int FF=Find(i);
                int jj;
                for(jj=0;jj<N;jj++)
                {
                    int Fjj=Find(jj);
                    if(FF==Fjj)
                        if(R[jj][j]==1)
                            break;
                }
                if(jj!=N) g[i][j]=1;
            }
        int Ans=0;
        while(1)
        {
            int UUU=MaxMatch();
            if(UUU!=N) break;
            Ans++;
            for(int i=0;i<N;i++)
                if(cx[i]!=-1)
                    g[i][cx[i]]=0;
        }
        printf("%d
",Ans);
    }
    return 0;
}