这个题解写的不错,有图:传送门
我们相当于有p*q个点,每个点有不同的取值,需要不同的代价,然后相邻的点取值有一些限制。我们可以把每个点拆成r+1个点,(x,y,z)->(x,y,z+1)连边,容量为a[x][y][z],s向所有的(x,y,1)连边,容量为inf,所有的(x,y,r+1)向T连边,容量为inf,不考虑高度限制的话,当前图的最小割就是答案。然后我们怎么实现高度的限制呢?(x,y,z)->(x’,y’,z-D)连边,容量为inf。这样的边有什么用呢?这样的话如果你选择的相邻两个高度不合法,他就一定不能把原图割开了。因此此时的最小割一定是合法的。
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define N 70000
#define inf 0x3f3f3f3f
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int T=65000,n,m,z,a[42][42][42],D,h[N],num=1,lev[N];
int dx[]={0,0,1,-1},dy[]={1,-1,0,0};
struct edge{
int to,next,val;
}data[700000];
inline int id(int x,int y,int zz){
return (x-1)*40*40+(y-1)*40+zz;
}
inline void add(int x,int y,int val){
data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=0;
}
inline bool bfs(){
queue<int>q;memset(lev,0,sizeof(lev));
q.push(0);lev[0]=1;
while(!q.empty()){
int x=q.front();q.pop();
for(int i=h[x];i;i=data[i].next){
int y=data[i].to;if(lev[y]||!data[i].val) continue;
lev[y]=lev[x]+1;q.push(y);
}
}return lev[T];
}
inline int dinic(int x,int low){
if(x==T) return low;int tmp=low;
for(int i=h[x];i;i=data[i].next){
int y=data[i].to;if(lev[y]!=lev[x]+1||!data[i].val) continue;
int res=dinic(y,min(tmp,data[i].val));
if(!res) lev[y]=0;tmp-=res;data[i].val-=res;data[i^1].val+=res;
if(!tmp) return low;
}return low-tmp;
}
int main(){
// freopen("a.in","r",stdin);
n=read();m=read();z=read();D=read();
for(int i=1;i<=z;++i)
for(int j=1;j<=n;++j)
for(int k=1;k<=m;++k)
a[j][k][i]=read();
for(int i=1;i<=n;++i)
for(int j=1;j<=m;++j){
add(0,id(i,j,1),inf);add(id(i,j,z+1),T,inf);
for(int k=1;k<=z;++k){
add(id(i,j,k),id(i,j,k+1),a[i][j][k]);
if(k-D<1) continue;
for(int kk=0;kk<4;++kk){
int x=i+dx[kk],y=j+dy[kk];
if(x<1||x>n||y<1||y>m) continue;
add(id(i,j,k),id(x,y,k-D),inf);
}
}
}int ans=0;
while(bfs()) ans+=dinic(0,inf);
printf("%d\n",ans);
return 0;
}