题目大意:在一个P*Q的格子里填数,不同的数代价不同,要求相邻的格子差小于D,求最小代价
题解:离散变量模型
如果没有D的限制的话,按层拆点,连(x,y,z)->(x,y,z+1),流量为v(x,y,z),原点向第一层连边,第R+1层向汇点连边,权为INF,这样最小割就可以了
现在加入限制,即相邻的只能割[z-d,z+d],可以加入一些权为INF的边屏蔽掉不能割的边,大概就是加一些桥梁,使得割掉不在[z-d,z+d]中的边后流量还可以绕过去,实现上只需要相邻的,层数差为d的点连INF的边,这个理性理解一下
我的收获:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
const int M=40*40*40+5;
const int dx[]={0,1,-1,0,0};
const int dy[]={0,0,0,1,-1};
bool Exit;
int n,m,h,D,t,st,ed,cnt,vx;
int id[65][65][65];
int num[M],d[M],head[M],last[M];
struct edge{int to,nex,c;}e[500005];
void add(int i,int j,int w){e[t].to=j,e[t].c=w,e[t].nex=head[i],last[i]=head[i]=t++;}
void insert(int x,int y,int z){add(x,y,z),add(y,x,0);};
int dfs(int x,int in)
{
if(x==ed) return in;
int ans=0,t;
for(int i=last[x];i!=-1;last[x]=i=e[i].nex)
{
int v=e[i].to;
if(e[i].c&&d[v]==d[x]-1){
t=dfs(v,min(in-ans,e[i].c));
ans+=t;
e[i].c-=t; e[i^1].c+=t;
if(Exit||ans==in) return ans;
}
}
if(--num[d[x]]==0) Exit=1;
d[x]++,num[d[x]]++,last[x]=head[x];
return ans;
}
int ISAP()
{
Exit=0;int flow=0;
while(!Exit) flow+=dfs(st,INF);
return flow;
}
void build()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
insert(st,id[1][i][j],INF),insert(id[h+1][i][j],ed,INF);
for(int k=1;k<=h;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
scanf("%d",&vx),insert(id[k][i][j],id[k+1][i][j],vx);
for(int k=1;k<=h;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
for(int p=1;p<=4;p++)
if(id[k+D][i+dx[p]][j+dy[p]])//方便的方法,不过需要开大数组
insert(id[k+D][i+dx[p]][j+dy[p]],id[k][i][j],INF);
}
void work(){
cout<<ISAP()<<endl;
}
void init()
{
memset(head,-1,sizeof(head));
memset(last,-1,sizeof(last));
cin>>n>>m>>h>>D;
for(int k=1;k<=h+1;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
id[k][i][j]=++cnt;
st=0,ed=cnt+1;num[0]=ed+1;
build();
}
int main()
{
init();
work();
return 0;
}