bzoj-3144 [Hnoi2013]切糕

3144 [Hnoi2013]切糕

题目链接

题目大意
给出一个立方体（长宽高分别P,Q,R），按三维坐标每个点记作（x,y,z），点权为V(x,y,z)，对于x,y位置从上到下有z层，但只能选择一层，并记作F(x,y)。对于任意一对相邻的（x,y），|F(x1,y1)-F(x2,y2)|≤d，d是一个常数。求ΣV(x,y,F(x,y))最小值。

题解
小题很考验想法，是一道网络流的题目。
我们的每一堆都只能选一层，所以我们先把点权当成边权来建图，每个节点向上建边。

然后把每堆构成的链的头和尾分别和源、汇这样，每一层只用割一条边就够了。

接下来，我们要做的是满足约束条件：高度差不超过d。
其实只需要对每个点向它下方的第 d 个点建边就好了。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
#define INF 123123123
using namespace std;
const int maxn=45,maxm=600005,f[4][2]={{0,1},{1,0},{-1,0},{0,-1}},maxnn=maxn*maxn*maxn;
int n,m,h,d,s,t,tot=-1,num,cur[maxnn],q[maxnn],v[maxn][maxn][maxn],id[maxn][maxn][maxn],lnk[maxnn],nxt[maxm],son[maxm],dep[maxnn];
LL w[maxm];
int read()
{
    int ret=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=getchar();
    return ret*f;
}
void add(int x,int y,int W)
{
    son[++tot]=y;nxt[tot]=lnk[x];lnk[x]=tot;w[tot]=W;
    son[++tot]=x;nxt[tot]=lnk[y];lnk[y]=tot;w[tot]=0;
}
LL dfs(int x,LL mis)
{
    if (mis==0||x==t) return mis;
    LL add=0,d;
    for (int &i=cur[x],y=son[i];~i;y=son[i=nxt[i]])
      if (dep[y]==dep[x]+1&&(d=dfs(y,min(mis,w[i]))))
      {
          w[i]-=d;w[i^1]+=d;
          mis-=d;add+=d;
          if (mis<=0) return add;
      }
    return add;
}
bool bfs(int s,int t)
{
    memset(dep,0,sizeof dep);
    for (int i=1;i<=num;i++) cur[i]=lnk[i];
    int hea=0,til=1;dep[q[1]=s]=1;
    while (hea!=til)
    {
        int x=q[++hea%=maxnn];
        for (int i=lnk[x],y=son[i];~i;y=son[i=nxt[i]])
          if (w[i]>0&&dep[y]==0)
          {
              dep[y]=dep[x]+1;
              q[++til%=maxnn]=y;
          }
    }
    return dep[t]!=0;
}
LL dinic(int s,int t)
{
    LL ans=0;
    while (bfs(s,t)) ans+=dfs(s,INF);
    return ans;
}
int main()
{
    memset(id,0,sizeof id);
    memset(lnk,-1,sizeof lnk);
    n=read(),m=read(),h=read();d=read();
    for (int k=1;k<=h+1;k++)
     for (int i=1;i<=n;i++)
      for (int j=1;j<=m;j++)
      {
          id[i][j][k]=++num;
          if (k<=h) v[i][j][k]=read();
          if (k>1) add(id[i][j][k-1],id[i][j][k],v[i][j][k-1]);
      }
    s=++num;t=++num;
    for (int i=1;i<=n;i++)
     for (int j=1;j<=m;j++) add(s,id[i][j][1],INF),add(id[i][j][h+1],t,INF);
    for (int k=d+1;k<=h+1;k++)
     for (int i=1;i<=n;i++)
      for (int j=1;j<=m;j++)
       for (int p=0,gto;p<4;p++)
         if (gto=id[i+f[p][0]][j+f[p][1]][k-d])
           add(id[i][j][k],gto,INF);
    printf("%lld",dinic(s,t));
    return 0;
}