洛谷 P4016负载平衡问题【费用流】

题目描述

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

输入格式：

文件的第 1 行中有 1个正整数 n，表示有 n 个仓库。
第 2 行中有 n 个正整数，表示 n 个仓库的库存量。

输出格式：

输出最少搬运量。

说明

1001≤n≤100

---

题目分析

建立超级源点
向每个仓库连边，容量为仓库库存量，费用为0
保证每个仓库一开始的库存量

然后每两个相邻的仓库连边，容量为inf，费用为1
这里每一次搬运费用就加一
所以最后的搬运量就是总费用

最后每个仓库向超级汇点连边
容量为最后平均的库存量，费用为0
保证每个仓库的库存量相等且总和相等

注意仓库环状排布，第n个仓库与第1个仓库要连边
仓库可以互相搬运，图是无向的

---

#include<iostream>
#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int inf=1128481603;
int k,n;
int s,t;
struct node{int v,f,c,nxt;}E[1000010];
int head[100010],tot=1;
int dis[100010],vis[100010];
int incf[100010],pre[100010],sum;
int fee;

void add(int u,int v,int f,int c)
{
    E[++tot].nxt=head[u];
    E[tot].f=f;
    E[tot].v=v;
    E[tot].c=c;
    head[u]=tot;
}


bool bfs()
{
    memset(dis,67,sizeof(dis)); dis[s]=0;
    queue<int> q; q.push(s); incf[s]=inf;
    
    while(!q.empty())
    {
        int u=q.front();
        q.pop(); vis[u]=0;
        for(int i=head[u];i;i=E[i].nxt)
        {
            int v=E[i].v;
            if(E[i].f&&dis[v]>dis[u]+E[i].c)
            {
                dis[v]=dis[u]+E[i].c;
                incf[v]=min(incf[u],E[i].f); pre[v]=i;
                if(!vis[v]) q.push(v),vis[v]=1;
            }
        }
    }
    return dis[t]!=inf;
}

void dfs()
{
    int u=t;
    while(u!=s)
    {
        int i=pre[u];
        E[i].f-=incf[t]; E[i^1].f+=incf[t];
        u=E[i^1].v;
    }
    fee+=dis[t]*incf[t];
}

int main()
{
    n=read(); t=n+1;
    for(int i=1;i<=n;i++)
    {
        int f=read(); sum+=f;
        add(s,i,f,0);add(i,s,0,0);
    }
    sum/=n;
    for(int i=1;i<=n;i++)
    {
        add(i,t,sum,0);add(t,i,0,0);
        if(i!=n)
        {
            add(i,i+1,inf,1);add(i+1,i,0,-1);
            add(i+1,i,inf,1);add(i,i+1,0,-1);
        }
        else if(i==n)
        { 
            add(i,1,inf,1);add(1,i,0,-1);
            add(1,i,inf,1);add(i,1,0,-1);
        }
    }
    while(bfs())dfs();
    
    cout<<fee;
    return 0;
}