负载平衡问题【网络流24题】【最大流最小费用】

版权声明:https://blog.csdn.net/qq_41730082 https://blog.csdn.net/qq_41730082/article/details/88827166

题目链接


  注意题意:环形——不然怎么也算不明白为什么样例是那样的…… 

  发现这道题竟然是一道费用流,是因为移动每一个货物可以看成是挪动一次代价为1,这样子想就可以建边了,从源点出发到每个节点是每个节点的拥有数,而每个节点到汇点的容量是那个平均数。然后对于每个相邻的节点连接上边并且单位流的代价是1。


#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int S = 0, maxN = 107, maxE = 1e3 + 7;
int N, head[maxN], cnt, all, T, pre[maxN], Flow[maxN], dist[maxN];
bool inque[maxN];
queue<int> Q;
struct Eddge
{
    int nex, u, v, flow, cost;
    Eddge(int a=-1, int b=0, int c=0, int d=0, int f=0):nex(a), u(b), v(c), flow(d), cost(f) {}
}edge[maxE];
inline void addEddge(int u, int v, int flow, int cost)
{
    edge[cnt] = Eddge(head[u], u, v, flow, cost);
    head[u] = cnt++;
}
inline void _add(int u, int v, int flow, int cost) { addEddge(u, v, flow, cost); addEddge(v, u, 0, -cost); }
bool spfa()
{
    memset(pre, -1, sizeof(pre));   memset(dist, INF, sizeof(dist));    memset(inque, false, sizeof(inque));
    while(!Q.empty()) Q.pop();
    Q.push(S);  inque[S] = true;    Flow[S] = INF;  dist[S] = 0;
    while(!Q.empty())
    {
        int u= Q.front();   inque[u] = false;   Q.pop();
        for(int i=head[u], v, flow, cost; ~i; i=edge[i].nex)
        {
            v = edge[i].v;  flow = edge[i].flow;    cost = edge[i].cost;
            if(flow && dist[v] > dist[u] + cost)
            {
                dist[v] = dist[u] + cost;
                pre[v] = i;
                Flow[v] = min(Flow[u], flow);
                if(!inque[v])
                {
                    inque[v] = true;
                    Q.push(v);
                }
            }
        }
    }
    return pre[T] ^ -1;
}
int Out_Put()
{
    int ans = 0;
    while(spfa())
    {
        int now = T, last = pre[now];
        while(now)
        {
            edge[last].flow -= Flow[T];
            edge[last ^ 1].flow += Flow[T];
            now = edge[last].u;
            last = pre[now];
        }
        ans += Flow[T] * dist[T];
    }
    return ans;
}
inline void init()
{
    cnt = all = 0;  T = N + 1;
    memset(head, -1, sizeof(head));
}
int main()
{
    scanf("%d", &N);
    init();
    for(int i=1, e1; i<=N; i++)
    {
        scanf("%d", &e1);
        all += e1;
        _add(S, i, e1, 0);
        if(i > 1) _add(i, i - 1, INF, 1);
        else _add(i, N, INF, 1);
        if(i < N) _add(i, i + 1, INF, 1);
        else _add(i, 1, INF, 1);
    }
    all /= N;
    for(int i=1; i<=N; i++) _add(i, T, all, 0);
    printf("%d\n", Out_Put());
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41730082/article/details/88827166
今日推荐