luoguP3227 [HNOI2013]切糕
题意
下面变量意义:\(n:P,m:Q,K:R\)。
先考虑没有\(D\)限制怎么做,我们只需要对\(n*m\)个点的所有\(K\)个值找最小值的即可。
发现这可以转化成网络流最小割:
1.从\(S\)向每个\((i,j,1)\)连容量为\(inf\)的边。
2.从\((i,j,k)\ k\in[1,K-1]\)向\((i,j,k+1)\)连容量为\(a_{i,j,k}\)的边。
3.从\((i,j,K)\)向\(T\)连容量为\(a_{i,j,K}\)的边。
现在我们考虑怎么处理\(D\)的限制:
假设现在有两个相邻的点,我们将第一列中的点称为\(1,2,3...\),第二列的点称为\(1',2',3'...\)。
我们考虑第一列割了\(i\),那么第二列的选择只有\([(i-D)',(i+D)']\),我们可以从\(i\)向\((i-D)'\)连一条容量为\(inf\)的边,这是我们发现如果割了\((i-D)'\)往前的边,仍存在一条从\(S->i->(i-D)'->T\)的路径,这是我们解决了不小于\((i-D)'\)的限制,因为是对称的,我们也会从\((i+D)'\)向\(i\)连一条边,于是我们也满足了不大于\((i+D)'\)的限制。
如图:
code:
#include<bits/stdc++.h>
using namespace std;
const int maxn=50;
const int maxl=125010;
const int inf=1e9;
const int dx[]={-1,1,0,0};
const int dy[]={0,0,-1,1};
int n,m,K,D,cnt=1,S,T,tot;
int head[maxl],cur[maxl],dep[maxl];
int a[maxn][maxn][maxn],id[maxn][maxn][maxn];
struct edge{int to,nxt,flow;}e[maxl<<5];
inline void add(int u,int v,int w)
{
e[++cnt].nxt=head[u];
head[u]=cnt;
e[cnt].to=v;
e[cnt].flow=w;
}
inline void addflow(int u,int v,int w){add(u,v,w),add(v,u,0);}
inline int read()
{
char c=getchar();int res=0,f=1;
while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9')res=res*10+c-'0',c=getchar();
return res*f;
}
inline bool bfs()
{
memset(dep,0,sizeof(dep));
for(int i=S;i<=T;i++)cur[i]=head[i];
queue<int>q;
q.push(S);dep[S]=1;
while(!q.empty())
{
int x=q.front();q.pop();
for(int i=head[x];i;i=e[i].nxt)
{
int y=e[i].to;
if(dep[y]||e[i].flow<=0)continue;
dep[y]=dep[x]+1;q.push(y);
}
}
return dep[T]>0;
}
int dfs(int x,int lim)
{
if(x==T||lim<=0)return lim;
int res=lim;
for(int i=cur[x];i;i=e[i].nxt)
{
cur[x]=i;
int y=e[i].to;
if(dep[y]!=dep[x]+1||e[i].flow<=0)continue;
int tmp=dfs(y,min(res,e[i].flow));
if(tmp<=0)dep[y]=0;
res-=tmp;
e[i].flow-=tmp,e[i^1].flow+=tmp;
if(res<=0)break;
}
return lim-res;
}
inline int Dinic()
{
int res=0;
while(bfs())res+=dfs(S,inf);
return res;
}
int main()
{
//freopen("test.in","r",stdin);
//freopen("test.out","w",stdout);
n=read(),m=read(),K=read(),D=read();
for(int k=1;k<=K;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
a[i][j][k]=read(),id[i][j][k]=++tot;
S=0,T=tot+1;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
addflow(S,id[i][j][1],inf);
for(int k=1;k<K;k++)addflow(id[i][j][k],id[i][j][k+1],a[i][j][k]);
addflow(id[i][j][K],T,a[i][j][K]);
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
for(int k=1;k<=K;k++)
{
if(k<=D)continue;
for(int u=0;u<4;u++)
{
int x=i+dx[u],y=j+dy[u];
if(x<1||x>n||y<1||y>m)continue;
addflow(id[i][j][k],id[x][y][k-D],inf);
}
}
printf("%d",Dinic());
return 0;
}