BZOJ3894: 文理分科

题目链接

【分析】

• 很显然的最小割。首先很容易想到将文理分成两个集合构图。但是一开始一直没出具体方法，看网上大神题解也不太理解。还好身边有位大爷抬了我一手，也让我对最小割有了更深刻的理解。
• 如果某个集合内的人都学文会获得一个满意度，那么就新加一个点，将集合内的所有人向这个点连流量为正无穷的边，再从这个点向T连一条流量为都选理满意度的边，表示集合内任意一个人学理都要把这个点与T的边割掉。
• 可能不太好理解，先看图：
  
  我们将中间一部分提出来：
  
  假设集合V={a,b,c,d,e};则V中每条边权代表第i人选文的获益值。V–>x得边显然是不能被割掉的，所以流量为INF；x–>T边权为这些人同时选理的获益值。
  若全部选理，则割掉V中所有边，若有人选文，则割掉边x–>T。
• 同理选文反过来即可。

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 30005
#define M 280005
#define INF 1000000001
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,cnt=1,S,T,ans;
int b[M],p[N],nextedge[M],w[M],cur[N];
int Level[N];
int dx[]={0,0,1,-1,0};
int dy[]={1,-1,0,0,0};

void Add(int x,int y,int z)
{
    cnt++;
    b[cnt]=y;
    nextedge[cnt]=p[x];
    p[x]=cnt;
    w[cnt]=z;
}

void Anode(int x,int y,int z){
    Add(x,y,z);Add(y,x,0);
}

void Input_Init()
{
    n=read(),m=read();T=3*n*m+1;
    static int x,num,xx,yy;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        x=read();ans+=x;
        Anode(0,(i-1)*m+j,x);
    } 
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        x=read();ans+=x;
        Anode((i-1)*m+j,T,x);
    }
    num=n*m;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        x=read();++num;
        Anode(0,num,x);ans+=x;
        for(int k=0;k<5;k++)
        {
            xx=i+dx[k],yy=j+dy[k];
            if(xx<1||yy<1||xx>n||yy>m) continue;
            Anode(num,(xx-1)*m+yy,INF);
        }
    }
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        x=read();++num;ans+=x;
        for(int k=0;k<5;k++)
        {
            xx=i+dx[k],yy=j+dy[k];
            if(xx<1||yy<1||xx>n||yy>m) continue;
            Anode((xx-1)*m+yy,num,INF);
        }
        Anode(num,T,x);
    }
}

bool Bfs()
{
    queue<int>q;
    q.push(S);
    for(int i=0;i<=T;i++) Level[i]=0;
    Level[S]=1;
    while(!q.empty())
    {
        int k=q.front();q.pop();
        for(int i=p[k];i;i=nextedge[i])
        {
            int v=b[i],f=w[i];
            if(!Level[v]&&f)
            {
                Level[v]=Level[k]+1;
                q.push(v);
            }
        }
    }
    return Level[T];
}

int Dfs(int x,int maxf)
{
    if(x==T||!maxf) return maxf;
    int rtn=0;
    for(int i=cur[x];i&&maxf>rtn;i=nextedge[i])
    {
        int v=b[i],f=w[i];
        if(Level[v]==Level[x]+1&&f)
        {
            f=Dfs(v,min(maxf-rtn,f));
            w[i]-=f;w[i^1]+=f;
            if(w[i]>0) cur[x]=i;
            rtn+=f;
        }
    }
    if(!rtn) Level[x]=0;
    return rtn;
}

void Dinic()
{
    while(Bfs())
    {
        for(int i=0;i<=T;i++) cur[i]=p[i];
        ans-=Dfs(S,INF);
    }   
    printf("%d\n",ans);
}

int main()
{
    Input_Init();
    Dinic();
    return 0;
}