[codeVS1916] 负载平衡问题（最小费用流，拆点）

题目链接：http://codevs.cn/problem/1916/

n个点之间可以相互传递值，每传递一次步骤+1。最终希望所有点的值相同，问最少步骤。

最终所有值应当是平均数avg，如何在两个相邻点之间传输值？可以先拆点，再建边：

将每一个点拆成两个点（两个点集分别记为①和②），同点②到①之间的容量为inf，费用为0。

拆完点后，按照要求每一个①点连向相邻两个点的②点，容量为inf，费用为1。

需要注意的是，保证移动完每一步可以判断是否能直接流向汇点，所以要在①点集合中向汇点建边，容量为avg，费用为0。

这里建图建挫了，大致是这样：

跑最小费用流就可以了。

 PS:这么看来BNU校赛的A题也可以这么建模（虽然贪心过的）。

附链接，可以思考：https://www.bnuoj.com/v3/contest_show.php?cid=9056#problem/A

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef struct Node {
    int u, v, next;
    LL c, w;
}Node;
const int maxn = 210;
const int maxm = 4010;
const LL mod = 0x3f3f3f3fLL;
const LL inf = (1LL<<55);
int tot, head[maxn];
LL dist[maxn];
LL cost, flow;
Node e[maxm];
int pre[maxn];
bool visit[maxn];
queue<int> Q;
int S, T, N;

void adde(int u, int v, LL c, LL w) {
    e[tot].u = u; e[tot].v = v; e[tot].c = c; e[tot].w = w; e[tot].next = head[u]; head[u] = tot++;
    e[tot].u = v; e[tot].v = u; e[tot].c = 0; e[tot].w = -w; e[tot].next = head[v]; head[v] = tot++;
}
bool spfa(int s, int t, int n) {
    int i;
    for(i = 0; i <= n; i++) {
        dist[i] = inf;
        visit[i] = 0;
        pre[i] = -1;
    }
    while(!Q.empty()) Q.pop();
    Q.push(s);
    visit[s] = true;
    dist[s] = 0;
    pre[s] = -1;
    while(!Q.empty()) {
        int u = Q.front();
        visit[u] = false;
        Q.pop();
        for(int j = head[u]; j != -1; j = e[j].next) {
            if(e[j].c > 0 && dist[u] + e[j].w < dist[e[j].v]) {
                dist[e[j].v] = dist[u] + e[j].w;
                pre[e[j].v] = j;
                if(!visit[e[j].v]) {
                    Q.push(e[j].v);
                    visit[e[j].v] = true;
                }
            }
        }
    }
    if(dist[t] == inf) return false;
    else return true;
}
LL ChangeFlow(int t) {
    LL det = mod;
    int u = t;
    while(~pre[u]) {
        u = pre[u];
        det = min(det, e[u].c);
        u = e[u].u;
    }
    u = t;
    while(~pre[u]) {
        u = pre[u];
        e[u].c -= det;
        e[u ^ 1].c += det;
        u = e[u].u;
    }
    return det;
}
LL MinCostFlow(int s, int t, int n) {
    LL mincost, maxflow;
    mincost = maxflow = 0;
    while(spfa(s, t, n)) {
        LL det = ChangeFlow(t);
        mincost += det * dist[t];
        maxflow += det;
    }
    cost = mincost;
    flow = maxflow;
    return mincost;
}

int n;
int a[maxn];

int main() {
    LL avg = 0;
    scanf("%d", &n);
    for(int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        avg += a[i];
    }
    avg /= n;
    tot = 0;
    memset(head, -1, sizeof(head));
    S = 0, T = 2 * n + 1; N = T + 1;
    for(int i = 1; i <= n; i++) {
        adde(0, i, a[i], 0);
        adde(i+n, i, inf, 0);
        adde(i, T, (LL)avg, 0);
    }
    for(int i = 1; i < n; i++) adde(i, i+n+1, inf, 1);
    adde(n, n+1, inf, 1);
    for(int i = 2; i <= n; i++) adde(i, i+n-1, inf, 1);
    adde(1, 2*n, inf, 1);
    cout << MinCostFlow(S, T, N) << endl;
    return 0;
}