BZOJ_3894_文理分科&&BZOJ_2127_happiness_最小割

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 30050
#define M 500050
#define S (n*m+1)
#define T (n*m+2)
#define p(i,j) ((i-1)*m+j)
#define inf 100000000
int head[N],to[M<<1],nxt[M<<1],cnt=1,n,m,dep[N],Q[N],l,r,flow[M<<1],sum,a[105][105],b[105][105];
int tx[]={0,1,-1,0};
int ty[]={1,0,0,-1};
inline void add(int u,int v,int f) {
	to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; flow[cnt]=f;
	to[++cnt]=u; nxt[cnt]=head[v]; head[v]=cnt; flow[cnt]=0;
}
bool bfs() {
	memset(dep,0,sizeof(dep));
	int i; l=r=0; Q[r++]=S; dep[S]=1;
	while(l<r) {
		int x=Q[l++];
		for(i=head[x];i;i=nxt[i]) {
			if(!dep[to[i]]&&flow[i]) {
				dep[to[i]]=dep[x]+1;
				if(to[i]==T) return 1;
				Q[r++]=to[i];
			}
		}
	}
	return 0;
}
int dfs(int x,int mf) {
	// puts("fgvfiugv");
	if(x==T) return mf;
	int i,nf=0;
	for(i=head[x];i;i=nxt[i]) {
		if(dep[to[i]]==dep[x]+1&&flow[i]) {
			int tmp=dfs(to[i],min(mf-nf,flow[i]));
			//if(!tmp) dep[to[i]]=0;
			nf+=tmp;
			flow[i]-=tmp;
			flow[i^1]+=tmp;
			if(nf==mf) break;
		}
	}
	return nf;
}
void dinic() {
	int f;
	while(bfs()) while(f=dfs(S,inf)) sum-=f;
	printf("%d\n",sum);
}
int main() {
	scanf("%d%d",&n,&m);
	int i,j,x,k;
	for(i=1;i<=n;i++) {
		for(j=1;j<=m;j++) {
			scanf("%d",&x);sum+=x;
			add(S,p(i,j),x);
		}
	}
	for(i=1;i<=n;i++) {
		for(j=1;j<=m;j++) {
			scanf("%d",&x);sum+=x;
			add(p(i,j),T,x);
		}
	}
	for(i=1;i<=n;i++) {
		for(j=1;j<=m;j++) {
			scanf("%d",&a[i][j]);sum+=a[i][j];
		}
	}
	for(i=1;i<=n;i++) {
		for(j=1;j<=m;j++) {
			scanf("%d",&b[i][j]);sum+=b[i][j];
		}
	}
	int tot=n*m+2;
	for(i=1;i<=n;i++) {
		for(j=1;j<=m;j++) {
			tot+=2;
			add(S,tot-1,a[i][j]);
			add(tot,T,b[i][j]);
			add(tot-1,p(i,j),inf);
			add(p(i,j),tot,inf);
			for(k=0;k<4;k++) {
				int di=i+tx[k],dj=j+ty[k];
				if(di>=1&&di<=n&&dj>=1&&dj<=m) {
					add(tot-1,p(di,dj),inf);
					add(p(di,dj),tot,inf);
				}
			}
		}
	}
	dinic();
}
/*
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 
*/