BJ模拟 Mortal Kombat【二分图匹配+tarjan】

题目大意:

给一张n个黑点,m个白点的二分图,问对于任意一对相连的黑白点对(i,j),判断边(i, j)是否能成为该二分图最大匹配的匹配边,能输出0,否则输出1。
n<=300,m<=1500,n<=m

解题思路:

先用匈牙利算法求一次最大匹配,则匹配边一定可以,所以就是判断非匹配边是否可以替换匹配边。
分两种情况讨论:
1.在最大匹配中的点对是否可以互相切换。
我们将匹配边从左向右连,非匹配边从右向左连。对新的图跑Tarjan,若左边的点i与右边的点j处于同一个强联通分量中,边(i, j)一定能成为最大匹配的匹配边(理由是一个强联通分量必定是偶数条边,可以从该点出发又回到该点,而这个路线恰好是匹配边-非匹配边-匹配边……,按顺序切换即可)。

2.一个白点i不在匹配中,但其实它连的边(i,j)都可以切换原匹配中的边(j,mat[j]), 而且切换后mat[j]又空出,一直递归处理即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=305,M=1505;
int n,m,vt,a[N][M],b[N][M],vis[M],mat[N+M];
int tot,first[N+M],nxt[N*M],to[N*M];
int num,top,idx,dfn[N+M],low[N+M],stk[N+M],bl[N+M],exist[N+M];
bool find(int i)
{
    for(int j=1;j<=m;j++)if(a[i][j]&&vis[j]!=vt)
    {
        vis[j]=vt;
        if(!mat[j+n]||mat[j+n]&&find(mat[j+n]))
        {mat[i]=j,mat[j+n]=i;return true;}
    }
    return false;
}
void add(int x,int y)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y;
}
void tarjan(int u)
{
    dfn[u]=low[u]=++idx;
    stk[++top]=u,exist[u]=1;
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];
        if(!dfn[v])tarjan(v),low[u]=min(low[u],low[v]);
        else if(exist[v])low[u]=min(low[u],dfn[v]);
    }
    if(dfn[u]==low[u])
    {
        ++num;
        while(stk[top]!=u)bl[stk[top]]=num,exist[stk[top--]]=0;
        bl[stk[top]]=num,exist[stk[top--]]=0;
    }
}
void change(int j)
{
    vis[j]=vt;
    for(int i=1;i<=n;i++)if(a[i][j])
    {
        b[i][j]=0;
        if(vis[mat[i]]!=vt)change(mat[i]);
    }
}
int main()
{
    //freopen("lx.in","r",stdin);
    //freopen("lx.out","w",stdout);
    n=getint(),m=getint();
    for(int i=1;i<=n;getchar(),i++)
        for(int j=1;j<=m;j++)
            a[i][j]=getchar()-'0',b[i][j]=a[i][j]^1;
    for(int i=1;i<=n;i++)if((++vt,!find(i)))
    {
        for(int i=1;i<=n;putchar('\n'),i++)
            for(int j=1;j<=m;j++)putchar('1');
        return 0;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)if(a[i][j])
            mat[i]==j?add(i,j+n):add(j+n,i);
    for(int i=1;i<=n+m;i++)if(!dfn[i])tarjan(i);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)if(a[i][j])
            b[i][j]=(mat[i]==j||bl[i]==bl[j+n])?0:1;
    ++vt;
    for(int j=1;j<=m;j++)if(!mat[j+n])change(j);
    for(int i=1;i<=n;putchar('\n'),i++)
        for(int j=1;j<=m;j++)putchar('0'+b[i][j]);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/cdsszjj/article/details/80217888
bj
今日推荐