记录编号 285429 评测结果 AAAAAAAAAAAAAAAAAAAA
题目名称 [HNOI 2013]切糕 最终得分 100
用户昵称 GravatarTenderRun 是否通过 通过
代码语言 C++ 运行时间 0.407 s
提交时间 2016-07-22 15:43:37 内存使用 12.71 MiB
显示代码纯文本
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
const int maxn=50010;
const int maxm=1000010;
const int INF=1000000000;
int cnt,fir[maxn],to[maxm],nxt[maxm],cap[maxm];
int dis[maxn],gap[maxn],path[maxn],fron[maxn];
queue<int>q;

struct Net_Flow{
    int tot;
    void Init(int tot_){
        memset(dis,0,sizeof(dis));
        memset(gap,0,sizeof(gap));
        memset(fir,0,sizeof(fir));
        cnt=1;tot=tot_;
    }
    void add(int a,int b,int c){
        nxt[++cnt]=fir[a];
        cap[cnt]=c;
        fir[a]=cnt;
        to[cnt]=b;
    }
    void addedge(int a,int b,int c){
        add(a,b,c);add(b,a,0);
    }
    bool BFS(int S,int T){
        dis[T]=1;q.push(T);
        while(!q.empty()){
            int x=q.front();q.pop();
            for(int i=fir[x];i;i=nxt[i])
                if(!dis[to[i]]){
                    dis[to[i]]=dis[x]+1;
                    q.push(to[i]);
                }
        }
        return dis[S];
    }
    int Max_Flow(int S,int T){
        if(!BFS(S,T))return 0;
        for(int i=0;i<tot;i++)fron[i]=fir[i];
        for(int i=0;i<tot;i++)gap[dis[i]]+=1;
        int ret=0,p=S,f,Min;
        while(dis[S]<=tot){
            if(p==T){
                f=INF;
                while(p!=S){
                    f=min(f,cap[path[p]]);
                    p=to[path[p]^1];
                }ret+=f;p=T;
                while(p!=S){
                    cap[path[p]]-=f;
                    cap[path[p]^1]+=f;
                    p=to[path[p]^1];
                }
            }
            
            for(int &i=fron[p];i;i=nxt[i])
                if(cap[i]&&dis[to[i]]==dis[p]-1)
                    {path[p=to[i]]=i;break;}
            
            if(!fron[p]){Min=tot;
                if(--gap[dis[p]]==0)break;
                for(int i=fir[p];i;i=nxt[i])
                    if(cap[i])Min=min(Min,dis[to[i]]);
                gap[dis[p]=Min+1]+=1;fron[p]=fir[p];
                if(p!=S)p=to[path[p]^1];    
            }        
        }
        return ret;
    }
}ISAP;

int P,Q,R,D,S,T;
int ID(int i,int j,int k){
	if(!k)return 0;
	return (k-1)*P*Q+(i-1)*Q+j;
}
int gx[4]={0,-1,0,1};
int gy[4]={1,0,-1,0};
int main(){
    freopen("nutcake.in","r",stdin);
    freopen("nutcake.out","w",stdout);
    
    scanf("%d%d%d%d",&P,&Q,&R,&D);
	S=0;T=P*Q*R+1;ISAP.Init(T+1);
    for(int k=1;k<=R;k++)
		for(int i=1;i<=P;i++)
			for(int j=1,w;j<=Q;j++){
        		scanf("%d",&w);
        		ISAP.addedge(ID(i,j,k-1),ID(i,j,k),w);
				if(k-D>0){int x,y,z=k-D;
					for(int t=0;t<4;t++){
						x=i+gx[t];y=j+gy[t];
						if(x>0&&x<=P&&y>0&&y<=Q)
						ISAP.addedge(ID(i,j,k),ID(x,y,z),INF);
					}
				}
				if(k==R)ISAP.addedge(ID(i,j,k),T,INF);
    		}		
    printf("%d\n",ISAP.Max_Flow(S,T));
    return 0;
}