[距离限制模型 & 网络流] BZOJ3144 :[Hnoi2013] 切糕

就是APIO2017何昊天课件里讲的模型

跟这题一样

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

const int N=100010,inf=1<<29;

int n,m,q,cnt=1,G[N],cur[N];
struct edge{
  int t,nx,f;
}E[N*10];

int f[45][45][45];

#define X(x,y,z) ((z-1)*n*m+(x-1)*m+y)

inline void link(int x,int y,int f){
  //cerr<<'\t'<<x<<' '<<y<<' '<<f<<endl;
  E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt; E[cnt].f=f;
  E[++cnt].t=x; E[cnt].nx=G[y]; G[y]=cnt; E[cnt].f=0;
}

int d,dis[N],S,T;
queue<int> Q;

inline bool bfs(){
  for(int i=0;i<=T;i++) dis[i]=-1;
  while(!Q.empty()) Q.pop();
  Q.push(S); dis[S]=0;
  while(!Q.empty()){
    int x=Q.front(); Q.pop();
    for(int i=G[x];i;i=E[i].nx)
      if(dis[E[i].t]==-1 && E[i].f){
    dis[E[i].t]=dis[x]+1; Q.push(E[i].t);
    if(E[i].t==T) return true;
      }
  }
  return false;
}

int dfs(int x,int f){
  if(x==T || !f) return f;
  int used=0,w;
  for(int &i=cur[x];i;i=E[i].nx)
    if(E[i].f && dis[E[i].t]==dis[x]+1){
      w=dfs(E[i].t,min(f-used,E[i].f));
      E[i].f-=w; E[i^1].f+=w;
      if((used+=w)==f) return f;
    }
  if(!used) dis[x]=-1;
  return used;
}

int main(){
  scanf("%d%d%d%d",&n,&m,&q,&d);
  S=0; T=X(n,m,q)+1;
  for(int k=1;k<=q;k++)
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++){
    scanf("%d",&f[i][j][k]);
      }
  for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++){
      for(int k=1;k<=q;k++){
    link(k==1?S:X(i,j,k-1),X(i,j,k),f[i][j][k]);
    if(i<n && k+d<=q)
      link(X(i+1,j,k+d),X(i,j,k),inf);
    if(i<n && k>1 &&k-d-1>0)
      link(X(i,j,k-1),X(i+1,j,k-d-1),inf);
    if(j<m && k+d<=q)
      link(X(i,j+1,k+d),X(i,j,k),inf);
    if(j<m && k>1 && k-d-1>0)
      link(X(i,j,k-1),X(i,j+1,k-d-1),inf);
      }
      link(X(i,j,q),T,inf);
    }
  int ans=0;
  while(bfs()){
    for(int i=0;i<=T;i++) cur[i]=G[i];
    ans+=dfs(S,inf);
  }
  printf("%d\n",ans);
  return 0;
}