洛谷 P4016 负载平衡问题

题面

题意

给出一个环上的仓库,每个仓库中有一些货物,每个仓库内的货物可以搬到相邻仓库,要使每个仓库的货物相等,至少要移动多少货物.

方法

网络流的题目难于减图,无法在其中建出源点和汇点,但可以这样考虑,将每一个仓库与所有仓库货物的平均值比较,若货物过多,向超级源点连一条流量为多出货物,费用为0的边,若少于,向汇点连一条流量为缺少的货物,费用为0的边,再在相邻两个仓库间连流量无限,费用为1的边,这样最大流可以保证平衡货物,而费用流可以保证费用最小.

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define INF 0x3f3f3f3f
#define N 110
using namespace std;

int n,m,num[N],first[N*3],sum,s=0,t,bb=1,d[N],last[N],B[N],ans;
bool in[N];
struct Bn
{
    int to,next,quan,cst;
}bn[N*10];
queue<int>que;

int dfs(int now,int mn)
{
    if(now==s) return mn;
    int res;
    res=dfs(last[now],min(mn,bn[B[now]].quan));
    bn[B[now]].quan-=res;
    bn[B[now]^1].quan+=res;
    return res;
}

inline bool bfs()
{
    int p,q;
    for(;!que.empty();que.pop());
    memset(d,INF,sizeof(d));
    memset(in,0,sizeof(in));
    que.push(s);
    d[s]=0;
    for(;!que.empty();)
    {
        q=que.front();
        que.pop();
        in[q]=0;
        for(p=first[q];p!=-1;p=bn[p].next)
        {
            if(d[q]+bn[p].cst>=d[bn[p].to]||!bn[p].quan) continue;
            d[bn[p].to]=d[q]+bn[p].cst;
            last[bn[p].to]=q;
            B[bn[p].to]=p;
            if(!in[bn[p].to])
            {
                in[bn[p].to]=1;
                que.push(bn[p].to);
            }
        }
    }
    if(d[t]==INF) return 0;
    ans+=dfs(t,INF)*d[t];
    return 1;
}

inline void add(int u,int v,int w,int z)
{
    bb++;
    bn[bb].to=v;
    bn[bb].next=first[u];
    bn[bb].quan=w;
    bn[bb].cst=z;
    first[u]=bb;
}

int main()
{
    memset(first,-1,sizeof(first));
    int i,j;
    cin>>n;
    t=n*2+1;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&num[i]);
        sum+=num[i];
    }
    sum/=n;
    for(i=1;i<=n;i++)
    {
        if(num[i]>sum) add(i,t,num[i]-sum,0),add(t,i,0,0);
        else if(num[i]<sum) add(s,i,sum-num[i],0),add(i,s,0,0);
    }
    for(i=1;i<=n;i++)
    {
        if(i==n)
        {
            add(n,1,INF,1);
            add(1,n,0,-1);
            add(1,n,INF,1);
            add(n,1,0,-1);
        }
        else
        {
            add(i,i+1,INF,1);
            add(i+1,i,0,-1);
            add(i+1,i,INF,1);
            add(i,i+1,0,-1);
        }
    }
    for(;bfs(););
    cout<<ans;
}