要求每个竖着的列选一条边出来,而且选择的权值和最小,那就直接上最小割
具体的,我们把每一列的所有边依次连接成为一条s到t的路径,做最小割即可
考虑怎么满足d的限制,也就是选了一条边必须选与这列相邻的列的某个连续范围的边
那么就从这条边的两个端点分别连对应的可选区间的两端,这样就限制了最小割必须割选定范围内的边,否则就不能割开
Code:
#include<bits/stdc++.h>
using namespace std;
inline int read(){
int res=0,f=1;char ch=getchar();
while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
return res*f;
}
const int N=7e4,M=3e5+5,INF=1e9;
int vis[M<<1],head[N],nxt[M<<1],c[M<<1],tot=1;
inline void add(int x,int y,int z){
vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;c[tot]=z;
vis[++tot]=x;nxt[tot]=head[y];head[y]=tot;c[tot]=0;
}
int st[N];
int s,t;
inline bool bfs(){
memset(st,-1,sizeof(st));
queue<int>q;
q.push(s);st[s]=1;
while(!q.empty()){
int x=q.front();q.pop();
for(int i=head[x];i;i=nxt[i]){
int y=vis[i];
if(c[i] && st[y]==-1){
q.push(y);
st[y]=st[x]+1;
if(y==t) return 1;
}
}
}
return 0;
}
int dfs(int v,int flow){
if(v==t) return flow;
int res=flow,k;
for(int i=head[v];i && res;i=nxt[i]){
int y=vis[i];
if(c[i] && st[y]==st[v]+1){
k=dfs(y,min(res,c[i]));
if(!k) st[y]=-1;
c[i]-=k;c[i^1]+=k;res-=k;
if(!flow) break;
}
}
return flow-res;
}
inline int dinic(){
int flow,mxflow=0;
while(bfs())
while(flow=dfs(s,INF)) mxflow+=flow;
return mxflow;
}
int v[45][45][45];
int d,p,q,r;
int xx[]={1,0,-1,0,0},yy[]={0,-1,0,1,0};
inline int num(int a,int b,int c){return a*p*q+b*q+c;}
int main(){
p=read();q=read();r=read();
d=read();
for(int i=1;i<=r;i++)
for(int j=1;j<=p;j++)
for(int k=1;k<=q;k++) v[i][j][k]=read();
s=N-2,t=N-1;
for(int i=1;i<=p;i++)
for(int j=1;j<=q;j++)
add(s,num(0,i,j),INF);
for(int i=1;i<=r;i++)
for(int j=1;j<=p;j++)
for(int k=1;k<=q;k++)
add(num(i-1,j,k),num(i,j,k),v[i][j][k]);
for(int i=1;i<=p;i++)
for(int j=1;j<=q;j++)
add(num(r,i,j),t,INF);
for(int i=d;i<=r;i++)
for(int j=1;j<=p;j++)
for(int k=1;k<=q;k++)
for(int t=0;t<=4;t++){
int nx=j+xx[t],ny=k+yy[t];
if(nx<1 || nx>p || ny<1|| ny>q) continue;
add(num(i,j,k),num(i-d,nx,ny),INF);
}
int ans=dinic();
cout<<ans;
return 0;
}