bzoj千题计划137：bzoj [CQOI2014]危桥

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

#define N 100
#define M 3000

const int inf=1e9;

int n,a1,a2,an,b1,b2,bn;

char s[51][51];

int tot;
int front[N],nxt[M<<1],to[M<<1],val[M<<1],from[M<<1];
int lev[N],num[N];
int path[N];
int cur[N];
 
int src,decc;

void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c))  c=getchar(); 
    while(isdigit(c)) { x=x*10+c-‘0‘; c=getchar();  }
}

void add(int u,int v,int w)
{
    to[++tot]=v; nxt[tot]=front[u]; front[u]=tot; from[tot]=u; val[tot]=w;
    to[++tot]=u; nxt[tot]=front[v]; front[v]=tot; from[tot]=v; val[tot]=0;
}

void build()
{
    tot=1;
    memset(front,0,sizeof(front));
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
        {
            if(s[i][j]==‘O‘) add(i,j,2);
            else if(s[i][j]==‘N‘) add(i,j,inf);
        }
    src=0; decc=n+1;
    add(src,a1,an<<1);
    add(a2,decc,an<<1);
    add(src,b1,bn<<1);
    add(b2,decc,bn<<1);
}

bool bfs()
{
    queue<int>q;
    for(int i=src;i<=decc;++i) lev[i]=decc+1;
    q.push(decc);
    lev[decc]=0;
    int now,t;
    while(!q.empty())
    {
        now=q.front();
        q.pop();
        for(int i=front[now];i;i=nxt[i])
        {
            t=to[i];
            if(lev[t]==decc+1 && val[i^1]) 
            {
                lev[t]=lev[now]+1;
                q.push(t);
            }
        }
    }
    return lev[src]!=decc+1;
}
 
int augment()
{
    int now=decc,flow=inf;
    int i;
    while(now!=src)
    {
        i=path[now];
        flow=min(flow,val[i]);
        now=from[i];
    }
    now=decc;
    while(now!=src)
    {
        i=path[now];
        val[i]-=flow;
        val[i^1]+=flow;
        now=from[i];
    }
    return flow;
}
 
int isap()
{
    int flow=0;
    if(!bfs()) return 0;
    memset(num,0,sizeof(num));
    for(int i=src;i<=decc;++i) num[lev[i]]++,cur[i]=front[i];
    int now=src,t;
    while(lev[src]<=decc)
    {
        if(now==decc)
        {
            flow+=augment();
            now=src;
        }
        bool advanced=false;
        for(int i=cur[now];i;i=nxt[i])
        {
            t=to[i];
            if(lev[t]==lev[now]-1 && val[i])
            {
                advanced=true;
                path[t]=i;
                cur[now]=i;
                now=t;
                break;
            }
        }
        if(!advanced)
        {
            int mi=decc;
            for(int i=front[now];i;i=nxt[i])
                if(val[i]) mi=min(mi,lev[to[i]]);
            if(!--num[lev[now]]) break;
            num[lev[now]=mi+1]++;
            cur[now]=front[now];
            if(now!=src) now=from[path[now]];
        }
    }
    return flow;
}

int main()
{
    int A,B;
    while(scanf("%d",&n)!=EOF)
    {
        read(a1); read(a2); read(an);
        a1++; a2++;
        read(b1); read(b2); read(bn);
        b1++; b2++;
        for(int i=1;i<=n;++i) scanf("%s",s[i]+1);
        build();
        A=isap();
        if(A!=(an+bn)*2) 
        {
            puts("No");
            continue;
        }
        swap(b1,b2);
        build();
        B=isap();
        puts(A==B ? "Yes" : "No");
    }
}