loj6013「网络流 24 题」负载平衡

(http://www.elijahqi.win/2017/12/12/6013%E3%80%8C%E7%BD%91%E7%BB%9C%E6%B5%81-24-%E9%A2%98%E3%80%8D%E8%B4%9F%E8%BD%BD%E5%B9%B3%E8%A1%A1/%20%E2%80%8E)
题目描述

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

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

输出最少搬运量。
样例
样例输入

5
17 9 14 16 4

样例输出

11

数据范围与提示

1≤n≤100 1 \leq n \leq 100 1≤n≤100

我把一个点 拆成两个点 一边是富裕的要给别人值的点 一遍是不富裕的要从别人得到的点

然后先求个平均值 然后针对每个地方的球都减一下这个平均值 然后如果富裕那么就安放在左边 如果不富裕 那就安放在右边 然后如果这两个点是临近点 那么我左边和左边的点相邻代表转移过去（只是经过而已）如果我左边和右边在原图中相邻的点连接 那么代表我这些球是用来补充那个缺少的 这些边的费用都是1 边权都是真无穷 然后源点和左边连接的边权是s[i]费用为0 右边和汇点连接的费用是0 边权是s[i] 然后跑费用流即可 我好菜啊orz啥都不会惨啊orz

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define inf 0x3f3f3f3f
#define N 220
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0;char ch=gc();
    while(ch<'0'||ch>'9') ch=gc();
    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=gc();}
    return x;
}
struct node{
    int y,next,z,c;
}data[N*N];
int h[N],pre[N],s[N],n,m,T,f[N],path[N],num=1;bool flag[N];
inline void insert1(int x,int y,int z,int c){
    data[++num].y=y;data[num].next=h[x];data[num].z=z;data[num].c=c;h[x]=num;
    data[++num].y=x;data[num].next=h[y];data[num].z=0;data[num].c=-c;h[y]=num;
}
inline bool spfa(){
    queue<int>q;memset(flag,0,sizeof(flag));memset(f,0x3f,sizeof(f));flag[0]=1;f[0]=0;q.push(0);memset(pre,-1,sizeof(pre));
    while(!q.empty()){
        int x=q.front();q.pop();flag[x]=0;
        for (int i=h[x];i;i=data[i].next){
            int y=data[i].y,z=data[i].z,c=data[i].c;
            if (f[x]+c<f[y]&&z){
                f[y]=f[x]+c;pre[y]=x;path[y]=i;
                if(!flag[y]) flag[y]=1,q.push(y);
            }
        }
    }if(pre[T]==-1) return 0;else return 1;
}
int main(){
    freopen("loj.in","r",stdin);
    n=read();int sum=0;T=2*n+1;
    for (int i=1;i<=n;++i) s[i]=read(),sum+=s[i];sum/=n;
    for (int i=1;i<=n;++i) s[i]-=sum;
    for (int i=1;i<=n;++i){//1~n为富裕的点 n+1~2*n为需求的点 
        if(s[i]>0) insert1(0,i,s[i],0);else insert1(i+n,T,-s[i],0);
        int last=(i-1)==0?n:i-1,next=(i+1)==n+1?1:i+1;
        insert1(i,next,inf,1);insert1(i,next+n,inf,1);insert1(i,last,inf,1);insert1(i,last+n,inf,1); 
    }int ans=0;
    while(spfa()){
        int minn=inf,now=T;
        while(now){minn=min(minn,data[path[now]].z);now=pre[now];}now=T;
        while(now){ans+=data[path[now]].c*minn;data[path[now]].z-=minn;data[path[now]^1].z+=minn;now=pre[now];}
    }printf("%d",ans);
    return 0;
}