poj 2516 Minimum Cost【最大流最小費用流】
阿新 • • 發佈:2018-12-07
題目連結:http://poj.org/problem?id=2516;
題目大意:有 n 個商店,m 個倉庫,k 種商品,問滿足供貨商要求的最小花費是多少,如果不能滿足輸出-1;前 n 行為每個商店需要商品的數量,接著 m 行為每個倉庫能放的多少個每種商品,然後是 k 個 n*m 的矩陣,表示第 k 個物品從倉庫 i 到商店 j 的花費:
思路:每種商品求一下最小費用流,求k次;
我的程式碼源點是 0 ,匯點是 n+m+1,與源點連的是商店,與匯點相連的是倉庫;
#include<cstring> #include<string> #include<cstdio> #include<stdlib.h> #include<iostream> #include<algorithm> #include<math.h> #include<map> #include<vector> #include<stack> #define inf 0x3f3f3f3f #include<queue> #include<set> using namespace std; typedef long long ll; const int N=105; const int M=20005; //建圖 struct node { int v,f,cos,ne; } edge[M]; int head[N*2],pre[N*2],dist[N*2],vis[N*2]; int s,t,e; //讀資料 int ned[N][N],has[N][N],val[N][N][N]; int nsum[N],hsum[N]; void add_insert(int a,int b,int c,int d) { edge[e].v=b; edge[e].f=c; edge[e].cos=d; edge[e].ne=head[a]; head[a]=e++; } void add(int a,int b,int c,int d) { add_insert(a,b,c,d); add_insert(b,a,0,-d); } int spaf() { memset(pre,-1,sizeof(pre)); memset(dist,inf,sizeof(dist)); memset(vis,0,sizeof(vis)); queue<int>que; vis[s]=1; dist[s]=0; que.push(s); while(!que.empty()) { int u=que.front(); que.pop(); vis[u]=0; for(int i=head[u]; i!=-1; i=edge[i].ne) { int v=edge[i].v,f=edge[i].f,c=edge[i].cos; if(dist[v]>dist[u]+c&&f) { dist[v]=dist[u]+c; pre[v]=i; if(!vis[v]) { vis[v]=1; que.push(v); } } } } return pre[t]!=-1; } int MCMF() { int flow=0,cost=0; while(spaf()) { int mini=inf; for(int i=pre[t]; i!=-1; i=pre[edge[i^1].v]) mini=min(mini,edge[i].f); for(int i=pre[t]; i!=-1; i=pre[edge[i^1].v]) { edge[i].f-=mini; edge[i^1].f+=mini; cost+=edge[i].cos*mini; } flow+=mini; } return cost; } int main() { int n,m,k; while(~scanf("%d %d %d",&n,&m,&k)) { if(!n&&!m&&!k) break; s=0;t=n+m+1; memset(nsum,0,sizeof(nsum)); memset(hsum,0,sizeof(hsum)); int ans=0; for(int i=1;i<=n;i++) { for(int j=1;j<=k;j++) { scanf("%d",&ned[i][j]);//記錄商店需要的貨物的數量 nsum[j]+=ned[i][j]; } } for(int i=1;i<=m;i++) { for(int j=1;j<=k;j++) { scanf("%d",&has[i][j]);//記錄倉庫存放的貨物的數量 hsum[j]+=has[i][j]; } } for(int i=1;i<=k;i++) { for(int j=1;j<=n;j++) { for(int l=1;l<=m;l++) scanf("%d",&val[i][j][l]);//記錄第l種貨物從i到j的花費 } } int flag=1; for(int i=1;i<=k;i++) { if(nsum[i]>hsum[i]) { flag=0; break; } } if(!flag) { printf("-1\n"); continue; } for(int l=1;l<=k;l++) { memset(head,-1,sizeof(head)); e=0; //源點到商店 for(int i=1;i<=n;i++) add(s,i,ned[i][l],0); //倉庫到匯點 for(int i=1;i<=m;i++) add(n+i,t,has[i][l],0); //商店到倉庫 for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) add(i,n+j,min(ned[i][l],has[j][l]),val[l][i][j]); ans+=MCMF(); } printf("%d\n",ans); } return 0; }