最小割模型,我们构造一个P*Q*(R+1)的点阵,相邻的点连边(i,j,k) –>(i,j,k+1),边权为下面的点的点权(v[i][j][k])。源点st向第一层的点连边,边权为INF,最上面一层的点向汇点ed连边,边权为顶层点权。
这样就把取点变成了割边,割掉(i,j,k) –>(i,j,k+1)这条边就是选取(i,j,k)这个点(f(i,j)=k)。
再从(i,j,k) –>(i’,j’,k-D)连一条边,流量INF,就能保证相邻两个不超过D了。
由于最小割=最大流,跑最大流即可。
Code:
#include<cstdio>
#include<cstdlib>
#include<cstring>
const int fx[4]={0,-1,0,1};
const int fy[4]={-1,0,1,0};
const int INF=0x3f3f3f3f;
struct node{int x,y,c,next,other;}a[400010];
int first[200010],h[200010],q[200010],v[50][50][50],id[50][50][50];
int P,Q,R,D,ans(0),len,st,ed;
void read(int &sum)
{
char ch=getchar();sum=0;int f=0;
while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
f=((ch=='-')&&(ch=getchar()));
while(ch>='0'&&ch<='9') sum=sum*10+(ch-48),ch=getchar();
(f)&&(sum=-sum);
}
void ins(int x,int y,int c)
{
a[++len]=(node){x,y,c,first[x],len+1};first[x]=len;
a[++len]=(node){y,x,0,first[y],len-1};first[y]=len;
}
int min(int x,int y)
{
return x<y?x:y;
}
bool bfs()
{
memset(h,0,sizeof(h));h[st]=1;
int head=1,tail=2;q[1]=st;
while(head!=tail)
{
int x=q[head];
for(int i=first[x];i;i=a[i].next)
{
int y=a[i].y;
if(a[i].c>0 && h[y]==0)
{
h[y]=h[x]+1;
q[tail++]=y;
}
}
head++;
}
if(h[ed]==0) return false;
return true;
}
int dfs(int x,int flow)
{
if(x==ed) return flow;
int minf,tt=0;
for(int i=first[x];i;i=a[i].next)
{
int y=a[i].y;
if(a[i].c>0 && h[y]==h[x]+1 && tt<flow)
{
minf=dfs(y,min(a[i].c,flow-tt));
tt+=minf;
a[i].c-=minf;a[a[i].other].c+=minf;
}
}
if(tt==0) h[x]=0;
return tt;
}
void dinic()
{
while(bfs())
ans+=dfs(st,INF);
}
int main()
{
read(P);read(Q);read(R);
read(D);
int tot(0);memset(first,len=0,sizeof(first));
for(int i=1;i<=R;i++)
for(int j=1;j<=P;j++)
for(int k=1;k<=Q;k++)
{
read(v[i][j][k]);
id[i][j][k]=++tot;
}
st=tot+1;ed=st+1;
for(int i=1;i<=R;i++)
for(int j=1;j<=P;j++)
for(int k=1;k<=Q;k++)
{
if(i==1) ins(st,id[i][j][k],INF);
if(i==R) ins(id[i][j][k],ed,v[i][j][k]);
else ins(id[i][j][k],id[i+1][j][k],v[i][j][k]);
}
for(int i=D+1;i<=R;i++)
for(int j=1;j<=P;j++)
for(int k=1;k<=Q;k++)
for(int l=0;l<4;l++)
{
int tx=j+fx[l],ty=k+fy[l];
if(tx<0 || tx>P || ty<0 || ty>Q) continue;
ins(id[i][j][k],id[i-D][tx][ty],INF);
}
dinic();
printf("%d",ans);
}