【JZOJ3222】【HNOI2013】切糕

Description

经过千辛万苦小A 得到了一块切糕，切糕的形状是长方体，小A 打算拦腰将切糕切成两半分给小B。出于美观考虑，小A 希望切面能尽量光滑且和谐。于是她找到你，希望你能帮她找出最好的切割方案。

出于简便考虑，我们将切糕视作一个长P、宽Q、高R 的长方体点阵。我们将位于第z层中第x 行、第y 列上(1≤x≤P, 1≤y≤Q, 1≤z≤R)的点称为(x,y,z)，它有一个非负的不和谐值v(x,y,z)。一个合法的切面满足以下两个条件：

1. 与每个纵轴(一共有P*Q 个纵轴)有且仅有一个交点。即切面是一个函数f(x,y)，对于所有1≤x≤P, 1≤y≤Q,我们需指定一个切割点f(x,y),且1≤f(x,y)≤R。

2. 切面需要满足一定的光滑性要求，即相邻纵轴上的切割点不能相距太远。对于所有的1≤x,x’≤P 和1≤y,y’ ≤Q，若|x-x’|+|y-y’|=1，则|f(x,y)-f(x’,y’)| ≤D，其中D 是给定的一个非负整数。

可能有许多切面f 满足上面的条件，小A 希望找出总的切割点上的不和谐值最小的那个，即最小。

Data Constraint

100%的数据满足P,Q,R≤40，0≤D≤R，且给出的所有的不和谐值不超过1000。

Solution

这道题刚看到时想到了以前在讲网络流时讲到的一个模型。既然要最小我们就想最小割。
我们可以将这些点拆成R层，每次第i层的（x,y）向第i+1层的（x,y）连一条容量为第i层（x,y）代价的边。那么当你取了第i层的（x,y）时，是有限制要求的，要求与(x,y)相邻的点选的层数必须在[i-D,i+D]的范围，所以我们将向与(x,y)相邻的点的i-D层连一条容量为INF的边，(x,y)相邻的点的i+1+D层向第i+1层的（x,y）连一条容量为INF的边，这样就保证了在与（x,y）相邻的点的[i-D,i+D]层必须割掉一条边。如图：

Code

#include<iostream> 
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e6+5,mo=1e9+7;
const int f[4][2]={{0,1},{1,0},{-1,0},{0,-1}};
int first[maxn],last[maxn],next[maxn],value[maxn],dui[maxn],d[maxn],v[maxn];
int n,m,q,p,i,t,j,k,l,x,y,z,num,ans,fa[maxn],g[maxn];
void lian(int x,int y,int z,int k){
    last[++num]=y;next[num]=first[x];first[x]=num;value[num]=z;dui[num]=num+1;
    last[++num]=x;next[num]=first[y];first[y]=num;value[num]=k;dui[num]=num-1;
}
int bfs(){
    int x,i=0,j=1,t;memset(d,0,sizeof(d));d[0]=1;
    while (i<j){
        x=v[++i];
        for (t=first[x];t;t=next[t]){
            if (d[last[t]]||!value[t])continue;
            v[++j]=last[t];d[v[j]]=d[x]+1;fa[v[j]]=x;g[v[j]]=t;
        }
    }
    if (d[n*m*q+1]) return 1;return 0;
}
int dg(int x,int sum){
    if (x==n*m*q+1) return sum;
    int k,t,p=sum;
    for (t=first[x];t;t=next[t]){
        if (!value[t]||d[x]+1!=d[last[t]])continue;
        k=dg(last[t],min(p,value[t]));
        if (k){
            value[t]-=k;value[dui[t]]+=k;p-=k;
            if (!p)break;
        }
    }
    if (sum==p) d[x]=-1;
    return sum-p;
}
int pan(int z,int x,int y){
    return z*n*m+(x-1)*m+y;
}
int main(){
//  freopen("data.in","r",stdin);
    scanf("%d%d%d%d",&n,&m,&q,&p);
    for (i=1;i<=n;i++)
        for (j=1;j<=m;j++)
            lian(0,pan(0,i,j),mo,0),lian(pan(q,i,j),n*m*(q+1)+1,mo,0);
    for (k=1;k<=q;k++){
        for (i=1;i<=n;i++)
            for (j=1;j<=m;j++){
                scanf("%d",&x);
                lian(pan(k-1,i,j),pan(k,i,j),x,0);
                for (l=0;l<4;l++){
                    x=i+f[l][0];y=j+f[l][1];
                    if (!x||!y||x>n||y>m)continue;
                    lian(pan(k-1,i,j),pan(max(0,k-1-p),x,y),mo,0);
                }
            }
        }
    q++;
    while (bfs())ans+=dg(0,mo);
    printf("%d\n",ans);
}