PKU2516 Minimum Cost - 最小费用最大流

模板题，整理代码……

 

1. /*
2. PKU2516 Minimum Cost
3. */
5. #include <stdio.h>
6. #include <memory.h>
8. #define clr(a) memset(a,0,sizeof(a))
9. #define N 200
10. #define INF (1<<25)
12. int DualityEdmondsKarp(int g[][N],int w[][N],int n,int s,int t,int f[][N],int *minCost)
13. {
14.     int i,j,k,c,quit,flow=0,minw=0;
15.     int best[N],prev[N];
16.     for(i=0;i<n;i++)for(j=0;j<n;j++) {
17.         f[i][j]=0;
18.         if(g[i][j]) {
19.             g[j][i]=0; w[j][i]=-w[i][j];
20.         }
21.     }
22.     while(1) {
23.         for(i=0;i<n;i++)best[i]=INF;
24.         best[s]=0;
25.         do{
26.             quit=1;
27.             for(i=0;i<n;i++)if(best[i]<INF){
28.                 for(j=0;j<n;j++)
29.                 if(f[i][j]<g[i][j]&
30.                 best[i]+w[i][j]<best[j]) {
31.                     best[j]=best[i]+w[i][j];
32.                     prev[j]=i;
33.                     quit=0;
34.                 }
35.             }
36.         } while(!quit);
37.         if(best[t]>=INF)break ;
38.         for(c=INF,j=t;j!=s;j=i) {
39.             i=prev[j];
40.             if(c>g[i][j]-f[i][j])c=g[i][j]-f[i][j];
41.         }
42.         for(j=t;j!=s;j=i) {
43.             i=prev[j];
44.             f[i][j]+=c;
45.             f[j][i]=-f[i][j];
46.         }
47.         flow+=c;
48.         minw+=c*best[t];
49.     }
50.     *minCost = minw;
51.     return flow;
52. }
54. int a[N][N][N];
55. int w[N][N][N];
56. int f[N][N];
58. int main()
59. {
60.     int i,j,k;
61.     int n,m,h;
62.     int s,t,sum,all;
64.     while(scanf("%d%d%d",&n,&m,&h)!=EOF && n+m+h){
65.         //init
66.         clr(a); clr(w);
67.         s=n+m; t=s+1;
68.         sum=0; all=0;
69.         for(k=0;k<h;k++)
70.         for(i=0;i<n;i++)for(j=0;j<m;j++){
71.             a[k][j][i+m]=INF;
72.         }
73.         
74.         //input
75.         for(i=0;i<n;i++)for(k=0;k<h;k++){
76.             scanf("%d",&a[k][i+m][t]);
77.             sum+=a[k][i+m][t];
78.         }
79.         for(j=0;j<m;j++)for(k=0;k<h;k++){
80.             scanf("%d",&a[k][s][j]);
81.             all+=a[k][s][j];
82.         }
83.         for(k=0;k<h;k++)
84.         for(i=0;i<n;i++)for(j=0;j<m;j++)
85.             scanf("%d",&w[k][j][i+m]);
86.         //prejudge
87.         if(sum>all){
88.             puts("-1"); continue;
89.         }
90.         //output
91.         int cost=0,c;
92.         int total=0;
93.         for(k=0;k<h;k++){
94.             total+=DualityEdmondsKarp(a[k],w[k],n+m+2,s,t,f,&c);
95.             cost+=c;
96.         }
97.         if(total<sum) puts("-1");
98.         else printf("%d/n",cost);
99.     }
101.     return 0;
102. }