洛谷 P3227 [HNOI2013]切糕（最小割）

传送门

---

最小割模型，我们构造一个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);
}