bzoj 3894 文理分科
3894: 文理分科
Description
文理分科是一件很纠结的事情！（虽然看到这个题目的人肯定都没有纠
结过）
小P所在的班级要进行文理分科。他的班级可以用一个n*m的矩阵进行
描述，每个格子代表一个同学的座位。每位同学必须从文科和理科中选择
一科。同学们在选择科目的时候会获得一个满意值。满意值按如下的方式
得到：
1．如果第i行第秒J的同学选择了文科，则他将获得art[i][j]的满意值，如
果选择理科，将得到science[i][j]的满意值。
2．如果第i行第J列的同学选择了文科，并且他相邻（两个格子相邻当且
仅当它们拥有一条相同的边）的同学全部选择了文科，则他会更开
心，所以会增加same_art[i][j]的满意值。
3．如果第i行第j列的同学选择了理科，并且他相邻的同学全部选择了理
科，则增加same_science[i]j[]的满意值。
小P想知道，大家应该如何选择，才能使所有人的满意值之和最大。请
告诉他这个最大值。
Input
第一行为两个正整数：n，m
接下来n术m个整数，表示art[i][j]；
接下来n术m个整数．表示science[i][j]；
接下来n术m个整数，表示same_art[i][j]；
Output
输出为一个整数，表示最大的满意值之和
Sample Input
3 4
13 2 4 13
7 13 8 12
18 17 0 5
8 13 15 4
11 3 8 11
11 18 6 5
1 2 3 4
4 2 3 2
3 1 0 4
3 2 3 2
0 2 2 1
0 2 4 4
13 2 4 13
7 13 8 12
18 17 0 5
8 13 15 4
11 3 8 11
11 18 6 5
1 2 3 4
4 2 3 2
3 1 0 4
3 2 3 2
0 2 2 1
0 2 4 4
Sample Output
152
HINT
N,M<=100,读入数据均<=500
题解：
建立源点和汇点分别表示学文和学理。
每个节点p向源点s连一流量为art的边，向汇点t连一流量为science的边，新建两个节点p1，p2，分别表示都学文和都学理，连边s->p1,p1->p,p->p2,p2->t.
流量分别为same_art、same_science.
用所有的满意值减去最小割即为答案。
有一瞬间我突然傻傻的想不通为什么学文和都学文的可以分开连边，然后想到，只要都连向源点（或汇点）就可以了，因为对于一个点，最小割一定只包括到达源点（或汇点）之一的边。
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#define inf 2147483647
#define nn 40000 //因为是矩形，所以要平方后再乘3
#define mm 1200000
#define lo long long
using namespace std;
bool vis[nn];
int x[4]={1,-1,0,0},y[4]={0,0,1,-1};
int art[110][110],sci[110][110],sa[110][110],ss[110][110];
int nxt[mm],fir[nn],to[mm],flow[mm],dis[nn],dep[nn],q[nn],ans=0,sum,e=1,n,m,S,T; //之前s和其他变量重名了，所以改成了大写
int getc()
{
int ans=0,f=1;char ch=getchar();
while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)) {ans=ans*10+ch-'0';ch=getchar();}
return ans*f;
}
void add(int a,int b,int d)
{
nxt[++e]=fir[a];fir[a]=e;to[e]=b;flow[e]=d;
nxt[++e]=fir[b];fir[b]=e;to[e]=a;flow[e]=0;
}
bool il(int x,int y)
{
if(x>=1&&x<=n&&y>=1&&y<=m)
return 1;
return 0;
}
void addedge()
{
int zc,zc2;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
zc=(i-1)*m+j;
add(S,zc,art[i][j]);
add(zc,T,sci[i][j]);
add(S,zc+sum,sa[i][j]);
add(zc+sum,zc,sa[i][j]);
add(zc,zc+2*sum,ss[i][j]);
add(zc+2*sum,T,ss[i][j]);
for(int k=0;k<4;k++)
if(il(i+x[k],j+y[k]))
{
zc2=(i+x[k]-1)*m+j+y[k];
add(zc+sum,zc2,sa[i][j]);
add(zc2,zc+sum*2,ss[i][j]);
}
}
}
bool bfs()
{
int h=1,t=1,o;
q[1]=S;
while(h<=t)
{
o=q[h++];
for(int i=fir[o];i;i=nxt[i])
if(flow[i]&&!dep[to[i]])
{
dep[to[i]]=dep[o]+1;
q[++t]=to[i];
}
}
if(dep[T]) return 1;
return 0;
}
int maxflow(int s,int f)
{
if(!f||s==T) return f; //写成了return 0
int newflow,newans=0;
for(int i=fir[s];i;i=nxt[i])
if(dep[to[i]]==dep[s]+1&&flow[i])
{
newflow=maxflow(to[i],min(f,flow[i]));
f-=newflow;
flow[i]-=newflow;
flow[i^1]+=newflow;
newans+=newflow;
if(!f) break;
}
if(f>0)
dep[s]=-1; //神奇的优化
return newans;
}
int main()
{
n=getc();m=getc();sum=n*m;
S=0;T=3*sum+1; //源、汇点是某确定点的时候，不要忘记赋值了
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
art[i][j]=getc();
ans+=art[i][j];
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
sci[i][j]=getc();
ans+=sci[i][j];
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
sa[i][j]=getc();
ans+=sa[i][j];
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
ss[i][j]=getc();
ans+=ss[i][j];
}
addedge();
dep[S]=1;
while(bfs())
{
ans-=maxflow(S,inf);
for(int i=1;i<=sum*3+1;i++)
dep[i]=0;
dep[S]=1;
}
printf("%d",ans);
return 0;
}
