LA 6234 Tile Cut dinic

蛋疼的输入。。。。。

把源和每个W连一条容量为1的边，由于I只能用1次所以把I进行分层，若不对I进行分层则可能通过I的有多条流量。

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>
#define INF 0x3fffffff

using namespace std;
char str[100][100];
#define maxn 222222
#define INF 0x3fffffff
struct node
{
    int next,c,v;
}e[maxn];
int en,ed,st,q[maxn],n,m;
int dis[maxn],first[maxn],cur[maxn];
int save[100][100];
bool bfs()
{
    memset(dis,-1,sizeof(dis));
    dis[st]=0;
    int tail,head;
    tail=head=1;
    tail++;
    q[tail]=st;
    int v,u;
    while(head<tail)
    {
        head++;
        u=q[head];
        for(int i=first[u];i!=-1;i=e[i].next)
        {
            v=e[i].v;
            if(dis[v]!=-1) continue;
            if(e[i].c==0) continue;
            dis[v]=dis[u]+1;
            tail++;
            q[tail]=v;
            if(v==ed) return true;
        }
    }
    return false;
}
int dfs(int x,int mx)
{
    if(x==ed||mx==0) return mx;
    int f,flow=0,v,ret=0;
    for(int& i=cur[x];i!=-1;i=e[i].next)
    {
        v=e[i].v;
        if(dis[x]+1!=dis[v]) continue;
        if((f=dfs(v,min(mx,e[i].c))))
        {
            e[i].c-=f;
            e[i^1].c+=f;
            flow+=f;
            ret+=f;
            mx-=f;
            if(!mx) break;
        }
    }
    if(ret==0) dis[x]=-1;
    return flow;
}
int dinic()
{
    int tmp=0,maxflow=0;
    while(bfs())
    {
        for(int i=1;i<=ed;i++) cur[i]=first[i];
        while(tmp=dfs(st,INF)) maxflow+=tmp;
    }
    return maxflow;
}
void add(int a,int b,int c)
{
    e[en].v=b;
    e[en].next=first[a];
    e[en].c=c;
    first[a]=en;
    en++;
    e[en].v=a;
    e[en].next=first[b];
    e[en].c=0;
    first[b]=en;
    en++;
}
int dx[]={0,0,1,-1,0};
int dy[]={0,1,0,0,-1};
void build()
{
    int xx,yy;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(save[i][j]==1)
            {
                add(st,(i-1)*m+j,1);
            }
            if(save[i][j]==2)
            {
                add((i-1)*m+j,(i-1)*m+j+m*n,1);
                for(int k=1;k<=4;k++)
                {
                    xx=i+dx[k];
                    yy=j+dy[k];
                    if(xx<1||xx>n||yy<1||yy>m) continue;
                    if(save[xx][yy]==1)
                    {
                        add((xx-1)*m+yy,(i-1)*m+j,1);
                    }
                    if(save[xx][yy]==3)
                    {
                        add((i-1)*m+j+m*n,(xx-1)*m+yy,1);
                    }
                }
            }
            if(save[i][j]==3)
            {
                add((i-1)*m+j,ed,1);
            }
        }
    }
}
void init()
{
    memset(first,-1,sizeof(first));
    en=0;
    st=n*m*2+1;
    ed=st+1;
}
void work()
{
    init();
    build();
    printf("%d\n",dinic());
}
int main()
{
    
    while(true)
    {
        bool flag = true;
        n=1;
        while(gets(str[n]))
        {
            n++;
            if(!str[n-1][0])
            {
                n--;
                flag = false;
                break;
            }
        }
        n--;
        m=strlen(str[1]);
        for(int j=1;j<=n;j++)
        {
            for(int i=1;i<=m;i++)
            {
                if(str[j][i-1]=='W') save[j][i]=1;
                if(str[j][i-1]=='I') save[j][i]=2;
                if(str[j][i-1]=='N') save[j][i]=3;
            }
        }
        work();
        m=0,n=0;
        if(flag) break;
    }
    return 0;
}