传送门
网络流好题。
首先考虑没有限制的情况。
这个时候每个位置求解是独立的。
其实可以直接贪心求最小值。
但也可以用最小割来求。
大概建图就是这样的:
考虑加入限制,那么直接暴力枚举给两个不能同时选的节点连一条容量为
i
n
f
inf
inf的边就行了。
这样需要付出
i
n
f
inf
inf的代价才能让这两个节点同时被选。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=45*45*45+5;
int p,q,r,d,val[45][45][45],dx[5]={0,0,-1,1,0},dy[5]={1,-1,0,0,0};
struct edge{int v,c,next;};
struct Dinic{
int s,t,cnt,cur[N],first[N],d[N];
edge e[N*105];
inline void init(){cnt=-1,memset(first,-1,sizeof(first)),s=0,t=N-5;}
inline void addedge(int u,int v,int c){e[++cnt].v=v,e[cnt].c=c,e[cnt].next=first[u],first[u]=cnt;}
inline void add(int u,int v,int c){addedge(u,v,c),addedge(v,u,0);}
inline bool bfs(){
queue<int>q;
memset(d,-1,sizeof(d)),q.push(s),d[s]=0;
while(!q.empty()){
int x=q.front();
q.pop();
for(int i=first[x];~i;i=e[i].next){
int v=e[i].v;
if(!e[i].c||~d[v])continue;
d[v]=d[x]+1,q.push(v);
}
}
return ~d[t];
}
inline int dfs(int x,int f){
if(x==t||!f)return f;
int flow=f;
for(int&i=cur[x];~i;i=e[i].next){
int v=e[i].v;
if(!flow)break;
if(e[i].c&&d[v]==d[x]+1){
int tmp=dfs(v,min(flow,e[i].c));
if(!tmp)d[v]=-1;
e[i].c-=tmp,e[i^1].c+=tmp,flow-=tmp;
}
}
return f-flow;
}
inline int solve(){
int ret=0;
while(bfs())memcpy(cur,first,sizeof(cur)),ret+=dfs(s,0x3f3f3f3f);
return ret;
}
}dinic;
inline int read(){
int ans=0;
char ch=getchar();
while(!isdigit(ch))ch=getchar();
while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
return ans;
}
inline int idx(int a,int b,int c){return a*p*q+b*q+c;}
int main(){
p=read(),q=read(),r=read(),d=read(),dinic.init();
for(int i=1;i<=r;++i)for(int j=1;j<=p;++j)for(int k=1;k<=q;++k)val[i][j][k]=read();
for(int i=1;i<=p;++i)for(int j=1;j<=q;++j)dinic.add(dinic.s,idx(0,i,j),0x3f3f3f3f),dinic.add(idx(r,i,j),dinic.t,0x3f3f3f3f);
for(int i=1;i<=r;++i)for(int j=1;j<=p;++j)for(int k=1;k<=q;++k)dinic.add(idx(i-1,j,k),idx(i,j,k),val[i][j][k]);
for(int i=d;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 mx=dx[l]+j,my=dy[l]+k;
if(!mx||!my||mx>p||my>q)continue;
dinic.add(idx(i,j,k),idx(i-d,mx,my),0x3f3f3f3f);
}
cout<<dinic.solve();
return 0;
}