【BZOJ 3144】 [Hnoi2013]切糕

原来dinic没学好啊

竟然还有个重要优化！

  1 #include <cstdio>
  2 #include <iostream>
  3 #include <cstring>
  4 #include <algorithm>
  5 using namespace std;
  6 #define MAXN 60
  7 #define INF 10000000
  8 int P,Q,R,D,s,t;
  9 int id[MAXN][MAXN][MAXN],cnt=0;
 10 int xx[5]={0,0,0,1,-1};
 11 int yy[5]={0,1,-1,0,0};
 12 int init[MAXN][MAXN][MAXN];
 13 int team[1000000],head,tail;
 14 int d[100000];
 15 int g[1000000],num[2000000],nnext[2000000],flow[2000000],tot=2;
 16 inline void Add(int x,int y,int z)
 17 {
 18 //    cout<<x<<' '<<y<<' '<<z<<endl;
 19     nnext[tot]=g[x];
 20     g[x]=tot;
 21     num[tot]=y;
 22     flow[tot]=z;
 23     tot++;
 24     
 25     nnext[tot]=g[y];
 26     g[y]=tot;
 27     num[tot]=x;
 28     flow[tot]=0;
 29     tot++;
 30 }
 31 inline bool bfs()
 32 {
 33     head=tail=0;
 34     memset(d,0,sizeof(d)); 
 35     team[++tail]=s;d[s]=1;
 36     while(head<tail)
 37     {
 38         int x=team[++head];
 39         for(int i=g[x];i;i=nnext[i])
 40         if(d[num[i]]==0&&flow[i]>0)
 41         {
 42             d[num[i]]=d[x]+1;
 43             team[++tail]=num[i];
 44         }
 45     }
 46     if(d[t]==0) return false;
 47     return true;
 48 }
 49 int dfs(int x,int mmin)
 50 {
 51     if(x==t) return mmin;
 52     int tmp,f=0;
 53     for(int i=g[x];i;i=nnext[i])
 54     if(d[num[i]]==d[x]+1&&flow[i]>0&&(tmp=dfs(num[i],min(mmin,flow[i]))))
 55     {
 56         mmin-=tmp;
 57         f+=tmp;
 58         flow[i]-=tmp;
 59         flow[i^1]+=tmp;
 60         if(mmin==0) 
 61         return f;
 62     }
 63     
 64     
 65     
 66     
 67     
 68     
 69     
 70     
 71     
 72     
 73     
 74     
 75     
 76     if(f==0) d[x]=0;
 77     
 78     
 79     
 80     
 81     
 82     
 83     
 84     
 85     
 86     
 87     
 88     
 89     
 90     
 91     return f;
 92 }
 93 int main()
 94 {
 95     //cin>>P>>Q>>R; 
 96     scanf("%d %d %d",&P,&Q,&R); 
 97    // cin>>D; 
 98     scanf("%d",&D);
 99     for(int i=1;i<=R;i++)
100       for(int j=1;j<=P;j++)
101          for(int k=1;k<=Q;k++)
102              id[j][k][i]=++cnt,
103              scanf("%d",&init[j][k][i]);
104     s=++cnt;
105     t=++cnt;
106   //  cout<<cnt<<endl;
107     for(int i=1;i<=P;i++)
108         for(int j=1;j<=Q;j++)
109         {
110             id[i][j][0]=s;
111             id[i][j][R+1]=t;
112             init[i][j][R+1]=INF;
113             for(int k=1;k<=R+1;k++)
114             {
115                 Add(id[i][j][k-1],id[i][j][k],init[i][j][k]);
116                 if(k-D>=1&&k!=R+1)
117                     for(int l=1;l<=4;l++)
118                         if(i+xx[l]>=1&&i+xx[l]<=P&&j+yy[l]>=1&&j+yy[l]<=Q)
119                             Add(id[i][j][k],id[i+xx[l]][j+yy[l]][k-D],INF);         
120             }
121         }
122     int ans=0;
123     while(bfs()) ans+=dfs(s,INF);
124     //cout<<ans;
125     printf("%d
",ans); 
126     return 0;
127 }