nefu493负载平衡问题

负载平衡问题

G 公司有n 个沿铁路运输线环形排列的仓库，每个仓库存储的货物数量不等。如何用最少搬运量可以使n 个仓库的库存数量相同。搬运货物时，只能在相邻的仓库之间搬运。
    对于给定的n 个环形排列的仓库的库存量，编程计算使n 个仓库的库存数量相同的最少搬运量。

【问题分析】
转化为供求平衡问题，用最小费用最大流解决。
【建模方法】
首先求出所有仓库存货量平均值，设第i个仓库的盈余量为A[i]，A[i] = 第i个仓库原有存货量 - 平均存货量。建立二分图，把每个仓库抽象为两个节点Xi和Yi。增设附加源S汇T。
1、如果A[i]>0，从S向Xi连一条容量为A[i]，费用为0的有向边。
2、如果A[i]<0，从Yi向T连一条容量为-A[i]，费用为0的有向边。
3、每个Xi向两个相邻顶点j，从Xi到Xj连接一条容量为无穷大，费用为1的有向边，从Xi到Yj连接一条容量为无穷大，费用为1的有向边。
求最小费用最大流，最小费用流值就是最少搬运量。
【建模分析】
计算出每个仓库的盈余后，可以把问题转化为供求问题。建立供求网络，把二分图X集合中所有节点看做供应节点，Y集合所有节点看做需求节点，在能一次搬运满足供需的Xi和Yj之间连接一条费用为1的有向边，表示搬运一个单位货物费用为1。另外还要在Xi与相邻的Xj之间连接边，表示货物可以暂时搬运过去，不立即满足需求，费用也为1。最大流满足了所有的盈余和亏损供求平衡，最小费用就是最少搬运量。

#include <cstdio>
using namespace std;
const int mm=111111;
const int mn=2222;
const int oo=1000000000;
int node,src,dest,edge;
int reach[mm],flow[mm],cost[mm],next[mm];
int head[mn],dis[mn],q[mn],p[mn];
bool vis[mn];
int a[mn];
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]=0;
    edge=0;
}
inline void addedge(int u,int v,int f,int c)
{
    reach[edge]=v,flow[edge]=f,cost[edge]=c,next[edge]=head[u],head[u]=edge++;
    reach[edge]=u,flow[edge]=0,cost[edge]=-c,next[edge]=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==mn)?l=0:1)
        for(i=head[u=q[l]],vis[u]=0;i>=0;i=next[i])
            if(flow[i]&&dis[v=reach[i]]>(tmp=dis[u]+cost[i]))
            {
                dis[v]=tmp;
                p[v]=i^1;
                if(vis[v])continue;
                vis[q[r++]=v]=1;
                if(r==mn)r=0;
            }
    return p[dest]>=0;
}
int SpfaFlow()
{
    int i,delta,ans=0;
    while(spfa())
    {
        for(i=p[dest],delta=oo;i>=0;i=p[reach[i]])
            delta=min(delta,flow[i^1]);
        for(i=p[dest];i>=0;i=p[reach[i]])
            flow[i]+=delta,flow[i^1]-=delta;
        ans+=delta*dis[dest];
    }
    return ans;
}
int main()
{
    int n,i,ave,sum,j;
    while(~scanf("%d",&n))
    {
        sum=0;
        prepare(n+n+2,0,n+n+1);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            sum+=a[i];
        }
        ave=sum/n;
        for(i=1;i<=n;i++)
            a[i]=a[i]-ave;
        for(i=1;i<=n;i++)
        {
            if(a[i]>=0)
                addedge(src,i,a[i],0);
            else
                addedge(i+n,dest,-a[i],0);
        }
        for(i=1;i<=n;i++)
        {
            if(i==1)
            {
                addedge(i,n+2,oo,1);
                addedge(i,n+n,oo,1);
                addedge(i,2,oo,1);
                addedge(i,n,oo,1);
            }
            else if(i==n)
                {
                    addedge(i,n+1,oo,1);
                    addedge(i,n+n-1,oo,1);
                    addedge(i,1,oo,1);
                    addedge(i,n-1,oo,1);
                }
                else
                {
                    addedge(i,i+1+n,oo,1);
                    addedge(i,i+1,oo,1);
                    addedge(i,i-1+n,oo,1);
                    addedge(i,i-1,oo,1);
                }
        }
        printf("%d\n",SpfaFlow());
    }
    return 0;
}