http://acm.nefu.edu.cn/JudgeOnline/problemshow.php?problem_id=492
description |
W公司有m个仓库和n 个零售商店。第i 个仓库有ai 个单位的货物;第j 个零售商店需要bj个单位的货物。货物供需平衡,即。从第i 个仓库运送每单位货物到第j 个零售商店的费用为Cij 。试设计一个将仓库中所有货物运送到零售商店的运输方案,使总运输费用最少。
对于给定的m 个仓库和n 个零售商店间运送货物的费用,计算最优运输方案和最差运输方案。
|
input |
|
output |
|
sample_input |
|
sample_output |
|
#include <iostream> #include <cstdio> using namespace std; const int oo=1e9;//无穷大 const int maxm=1111111;//边的最大数量,为原图的两倍 const int maxn=2222;//点的最大数量 int node,src,dest,edge;//node节点数,src源点,dest汇点,edge边数 int head[maxn],p[maxn],dis[maxn],q[maxn],vis[maxn];//head链表头,p记录可行流上节点对应的反向边,dis计算距离 struct edgenode { int to;//边的指向 int flow;//边的容量 int cost;//边的费用 int next;//链表的下一条边 } edges[maxm]; void prepare(int _node,int _src,int _dest); void addedge(int u,int v,int f,int c); bool spfa(); inline int min(int a,int b) { return a<b?a:b; } inline void prepare(int _node,int _src,int _dest) { node=_node; src=_src; dest=_dest; for (int i=0; i<node; i++) { head[i]=-1; vis[i]=false; } edge=0; } void addedge(int u,int v,int f,int c) { edges[edge].flow=f; edges[edge].cost=c; edges[edge].to=v; edges[edge].next=head[u]; head[u]=edge++; edges[edge].flow=0; edges[edge].cost=-c; edges[edge].to=u; edges[edge].next=head[v]; head[v]=edge++; } bool spfa() { int i,u,v,l,r=0,tmp; for (i=0; i<node; i++) dis[i]=oo; dis[q[r++]=src]=0; p[src]=p[dest]=-1; for (l=0; l!=r; ((++l>=maxn)?l=0:1)) { for (i=head[u=q[l]],vis[u]=false; i!=-1; i=edges[i].next) { if (edges[i].flow&&dis[v=edges[i].to]>(tmp=dis[u]+edges[i].cost)) { dis[v]=tmp; p[v]=i^1; if (vis[v]) continue; vis[q[r++]=v]=true; if (r>=maxn) r=0; } } } return p[dest]>=0; } int spfaflow() { int i,ret=0,delta; while (spfa()) { //按记录原路返回求流量 for (i=p[dest],delta=oo; i>=0; i=p[edges[i].to]) { delta=min(delta,edges[i^1].flow); } for (int i=p[dest]; i>=0; i=p[edges[i].to]) { edges[i].flow+=delta; edges[i^1].flow-=delta; } ret+=delta*dis[dest]; } return ret; } int a[10005],b[10005],sum[10005][10005]; int main() { int n,m; while(~scanf("%d%d",&n,&m)) { prepare(n+m+2,0,n+m+1); for(int i=1; i<=n; i++) { scanf("%d",&a[i]); addedge(src,i,a[i],0); } for(int i=1; i<=m; i++) { scanf("%d",&b[i]); addedge(i+n,dest,b[i],0); } for(int i=1; i<=n; i++) for(int j=1; j<=m; j++) { scanf("%d",&sum[i][j]); addedge(i,j+n,oo,sum[i][j]); } printf("%d\n",spfaflow()); prepare(n+m+2,0,n+m+1); for(int i=1; i<=n; i++) addedge(src,i,a[i],0); for(int i=1; i<=m; i++) addedge(i+n,dest,b[i],0); for(int i=1; i<=n; i++) for(int j=1; j<=m; j++) addedge(i,j+n,oo,-sum[i][j]); printf("%d\n",-spfaflow()); } return 0; }