BZOJ3894：文理分科——题解

http://www.lydsy.com/JudgeOnline/problem.php?id=3894

网络流，虽然我知道一个比我下面说的要简单的做法，但是我忘了（滑稽。

按照最小割的套路，与S有边表示选文，与T有边表示选理。

那么S对每个人连art边权，每个人对T连sci边权。

棘手的就是处理文科和理科之间的关系。

考虑对每个人新建同文理点，向与其相邻的人（包括他自己）连INF的边（注意方向别反了），S向同文点连sameart边，同理点向T连samesci边。

（自己画一画图大概就能感受到它的正确性了）

跑一遍最大流最小割，然后用总边权减掉最大流即可。

#include<cstdio>

#include<cmath>

#include<iostream>

#include<vector>

#include<cstring>

#include<algorithm>

#include<cctype>

using namespace std;

typedef long long ll;

const int N=1e5+;

const int M=N*;

const int INF=1e9;

inline int read(){

    int X=,w=;char ch=;

    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}

    while(isdigit(ch))X=(X<<)+(X<<)+(ch^),ch=getchar();

    return w?-X:X;

}

struct node{

    int nxt,to,w;

}edge[M];

int head[N],cnt=-,S,T;

inline void add(int u,int v,int w){

    edge[++cnt].to=v;edge[cnt].w=w;edge[cnt].nxt=head[u];head[u]=cnt;

    edge[++cnt].to=u;edge[cnt].w=;edge[cnt].nxt=head[v];head[v]=cnt;

}

int lev[N],cur[N],dui[N],a[][],b[][];

int da[][],db[][];

bool bfs(int m){

    int r=;

    for(int i=;i<=m;i++){

        lev[i]=-;

        cur[i]=head[i];

    }

    dui[]=S,lev[S]=;

    int u,v;

    for(int l=;l<=r;l++){

        u=dui[l];

        for(int e=head[u];e!=-;e=edge[e].nxt){

            v=edge[e].to;

            if(edge[e].w>&&lev[v]==-){

                lev[v]=lev[u]+;

                r++;

                dui[r]=v;

                if(v==T)return ;

            }

        }

    }

    return ;

}

int dinic(int u,int flow,int m){

    if(u==m)return flow;

    int res=,delta;

    for(int &e=cur[u];e!=-;e=edge[e].nxt){

        int v=edge[e].to;

        if(edge[e].w>&&lev[u]<lev[v]){

            delta=dinic(v,min(edge[e].w,flow-res),m);

            if(delta>){

                edge[e].w-=delta;

                edge[e^].w+=delta;

                res+=delta;

                if(res==flow)break;

            }

        }

    }

    if(res!=flow)lev[u]=-;

    return res;

}

int dx[]={,,,-,};

int dy[]={,,-,,};

int main(){

    memset(head,-,sizeof(head));

    int n=read(),m=read(),ans=;

    S=n*m*+,T=S+;

    for(int i=;i<=n;i++){

    for(int j=;j<=m;j++){

        a[i][j]=read();

        ans+=a[i][j];

    }

    }

    for(int i=;i<=n;i++){

    for(int j=;j<=m;j++){

        b[i][j]=read();

        ans+=b[i][j];

    }

    }

    for(int i=;i<=n;i++){

    for(int j=;j<=m;j++){

        da[i][j]=read();

        ans+=da[i][j];

    }

    }

    for(int i=;i<=n;i++){

    for(int j=;j<=m;j++){

        db[i][j]=read();

        ans+=db[i][j];

    }

    }

    for(int i=;i<=n;i++){

    for(int j=;j<=m;j++){

        int now=(i-)*m+j;

        add(S,now,a[i][j]);

        add(now,T,b[i][j]);

        add(S,now+n*m,da[i][j]);

        add(now+*n*m,T,db[i][j]);

        for(int k=;k<;k++){

        int nx=i+dx[k],ny=j+dy[k];

        if(nx<||nx>n||ny<||ny>m)continue;

        int pre=(nx-)*m+ny;

        add(now+n*m,pre,INF);

        add(pre,now+*n*m,INF);

        }

    }

    }

    while(bfs(T))ans-=dinic(S,INF,T);

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

    return ;

}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++