洛谷 - P2598 - 狼和羊的故事 - 最大流

https://www.luogu.org/problemnew/show/P2598

第一次写这种修篱笆的题目，上次好像晓阳dalao写了一个堵人的。好像还有能改变土地属性的。

#include<bits/stdc++.h>
using namespace std;
#define ll long long

/* dinic begin */

const int MAXN=;
const int MAXM=;
const int INF=0x3f3f3f3f;
struct Edge{
    int to,next,cap,flow;
}edge[MAXM];

int tol;
int head[MAXN];

void init(){
    tol=;
    memset(head,-,sizeof(head));
}

void addedge(int u,int v,int w){
    edge[tol].to=v;edge[tol].cap=w;edge[tol].flow=;
    edge[tol].next=head[u];head[u]=tol++;
    edge[tol].to=u;edge[tol].cap=;edge[tol].flow=;
    edge[tol].next=head[v];head[v]=tol++;
}

int Q[MAXN];
int dep[MAXN],cur[MAXN],sta[MAXN];
bool bfs(int s,int t,int n){
    int front=,tail=;
    memset(dep,-,sizeof(dep[])*(n+));
    dep[s]=;
    Q[tail++]=s;
    while(front<tail){
        int u=Q[front++];
        for(int i=head[u];i!=-;i=edge[i].next){
            int v=edge[i].to;
            if(edge[i].cap>edge[i].flow&&dep[v]==-){
                dep[v]=dep[u]+;
                if(v==t)
                    return true;
                Q[tail++]=v;
            }
        }
    }
    return false;
}

int dinic(int s,int t,int n){
    //n最后一个节点的编号的下一个编号
    int maxflow=;
    while(bfs(s,t,n)){
        for(int i=;i<n;i++)cur[i]=head[i];
        int u=s,tail=;
        while(cur[s]!=-){
            if(u==t){
                int tp=INF;
                for(int i=tail-;i>=;i--){
                    tp=min(tp,edge[sta[i]].cap-edge[sta[i]].flow);

                }
                maxflow+=tp;
                for(int i=tail-;i>=;i--){
                    edge[sta[i]].flow+=tp;
                    edge[sta[i]^].flow-=tp;
                    if(edge[sta[i]].cap-edge[sta[i]].flow==)
                        tail=i;
                }
                u=edge[sta[tail]^].to;

            }
            else if(cur[u]!=-&&edge[cur[u]].cap>edge[cur[u]].flow
                    &&dep[u]+==dep[edge[cur[u]].to]){
                sta[tail++]=cur[u];
                u=edge[cur[u]].to;
            }
            else{
                while(u!=s&&cur[u]==-){
                    u=edge[sta[--tail]^].to;
                }
                cur[u]=edge[cur[u]].next;
            }
        }
    }
    return maxflow;
}

/* dinic end */

int g[][];

int m,n;
int main(){
    init();
    scanf("%d%d",&n,&m);

    for(int i=;i<=n;i++){
        for(int j=;j<=m;j++){
            scanf("%d",&g[i][j]);
        }
    }

    int s=,t=n*m+;
    for(int i=;i<=n;i++){
        for(int j=;j<=m;j++){
            if(g[i][j]==){
                addedge(s,(i-)*m+j,INF);
            }
            else if(g[i][j]==){
                addedge((i-)*m+j,t,INF);
            }
        }
    }

    for(int i=;i<=n;i++){
        for(int j=;j<=m;j++){
            if(g[i][j]==||g[i][j]==){
                if(j->=&&g[i][j-]!=){
                    addedge((i-)*m+j,(i-)*m+j-,);
                    //addedge((i-1)*m+j-1,(i-1)*m+j,1);
                }
                if(i->=&&g[i-][j]!=){
                    addedge((i-)*m+j,(i-)*m+j,);
                    //addedge((i-2)*m+j,(i-1)*m+j,1);
                }
                if(j+<=m&&g[i][j+]!=){
                    addedge((i-)*m+j,(i-)*m+j+,);
                    //addedge((i-1)*m+j+1,(i-1)*m+j,1);
                }
                if(i+<=n&&g[i+][j]!=){
                    addedge((i-)*m+j,i*m+j,);
                    //addedge(i*m+j,(i-1)*m+j,1);
                }
                //羊可以在所有和他不同的格子修篱笆
            }
            /*else if(g[i][j]==0){
                //空地可以向任何方向修篱笆，但不需要向羊修
                if(j-1>=1&&g[i][j-1]!=1){
                    addedge((i-1)*m+j,(i-1)*m+j-1,1);
                }
                if(i-1>=1&&g[i][j]!=g[i-1][j]){
                    addedge((i-1)*m+j,(i-2)*m+j,1);
                }
                if(j+1<=m&&g[i][j]!=g[i][j+1]){
                    addedge((i-1)*m+j,(i-1)*m+j+1,1);
                }
                if(i+1<=n&&g[i][j]!=g[i+1][j]){
                    addedge((i-1)*m+j,i*m+j,1);
                }
            }*/
        }
    }

    int maxflow=dinic(s,t,t);

    printf("%d\n",maxflow);

}