1754: 负载平衡问题（最小费用最大流）

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

思路：贪心是很好想的，不过这道题用来练习网络流。。。

通过分析发现，求都减去平均数后，剩下的就是需要的或多出来的数量。把所有多出的仓库与源点连边，容量为多出的数量，费用0。把所有缺少的仓库与汇点建边，容量为缺少的数量，费用0。然后对每个点对两边的点建边，容量INF，费用1，就是每运一个货物花费1，跑最小费用最大流。

#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int N = 1e5 + 10;
int n, m, s, t, h[N], cnt, dis[N], inq[N], pre[N], last[N], flow[N];
struct node {
    int v, w, c, nt;
} no[N];
void add(int u, int v, int w, int c) {
    no[cnt] = node{v, w, c, h[u]};
    h[u] = cnt++;
    no[cnt] = node{u, 0, -c, h[v]};
    h[v] = cnt++;
}
int spfa() {
    memset(inq, 0, sizeof inq);
    memset(dis, INF, sizeof dis);
    dis[s] = 0, inq[s] = 1, flow[s] = INF;
    queue<int> q;
    q.push(s);
    while(!q.empty()) {
        int u = q.front();
        q.pop(), inq[u] = 0;
        for(int i = h[u]; ~i; i = no[i].nt) {
            int v = no[i].v;
            if(no[i].w > 0 && dis[v] > dis[u] + no[i].c) {
                dis[v] = dis[u] + no[i].c;
                pre[v] = u, last[v] = i;
                flow[v] = min(flow[u], no[i].w);
                if(!inq[v])
                    inq[v] = 1, q.push(v);
            }
        }
    }
    return dis[t] != INF;
}
ll mcmf() {
    ll ans = 0;
    while(spfa()) {
        for(int i = t; i != s; i = pre[i]) {
            no[last[i]].w -= flow[t];
            no[last[i] ^ 1].w += flow[t];
        }
        ans += 1ll * flow[t] * dis[t];
    }
    return ans;
}
int ap[N], sum;
int main() {
    memset(h, -1, sizeof h);
    scanf("%d", &n);
    for(int i = 1; i <= n; i++)
        scanf("%d", ap + i), sum += ap[i];
    int avg = sum / n;
    t = n + 1;
    for(int i = 1; i <= n; i++) {
        ap[i] = ap[i] - avg;
        if(ap[i] > 0)
            add(s, i, ap[i], 0);
        else
            add(i, t, -ap[i], 0);
        ap[i] = abs(ap[i]);
        if(i == 1)
            add(1, 2, INF, 1), add(1, n, INF, 1);
        else if(i == n)
            add(n, 1, INF, 1), add(n, n - 1, INF, 1);
        else
            add(i, i + 1, INF, 1), add(i, i - 1, INF, 1);
    }
    printf("%lld\n", mcmf());
    return 0;
}