题目描述
GG 公司有 nn 个沿铁路运输线环形排列的仓库,每个仓库存储的货物数量不等。如何用最少搬运量可以使 nn 个仓库的库存数量相同。搬运货物时,只能在相邻的仓库之间搬运。
输入输出格式
输入格式:
文件的第 11 行中有 11 个正整数 nn ,表示有 nn 个仓库。
第 22 行中有 nn 个正整数,表示 nn 个仓库的库存量。
输出格式:
输出最少搬运量。
输入输出样例
输入样例#1:
5
17 9 14 16 4
输出样例#1:
11
说明
1 leq n leq 1001≤n≤100
解题思路
第一种方法就是环形均分纸牌,和糖果传递一模一样。
第二种方法是网络流,建立超级源点S,S向比平均值大的连一条容量为a[i]-平均值,花费为0的边,建立超级源点T,比平均值小的点向T连一条容量为平均值-a[i],花费为0的边,然后每个点向它左右连一条容量为inf,花费为1的边,然后跑费用流即可。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int MAXN = 10005;
const int inf = 0x3f3f3f3f;
inline int rd(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)) {f=ch=='-'?0:1;ch=getchar();}
while(isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
return f?x:-x;
}
int n,a[MAXN],sum,head[MAXN],cnt=1,S,T;
int to[MAXN<<1],nxt[MAXN<<1],val[MAXN<<1],cost[MAXN<<1];
int incf[MAXN],pre[MAXN],dis[MAXN],ans;
bool vis[MAXN];
queue<int> Q;
inline void add(int bg,int ed,int w,int c){
to[++cnt]=ed,nxt[cnt]=head[bg],val[cnt]=w,cost[cnt]=c,head[bg]=cnt;
to[++cnt]=bg,nxt[cnt]=head[ed],val[cnt]=0,cost[cnt]=-c,head[ed]=cnt;
}
bool spfa(){
memset(vis,false,sizeof(vis));
memset(dis,0x3f,sizeof(dis));
while(Q.size()) Q.pop();
dis[S]=0,vis[S]=1,Q.push(S),incf[S]=inf;
while(Q.size()){
int x=Q.front();Q.pop();
for(register int i=head[x];i;i=nxt[i]){
int u=to[i];
if(val[i] && dis[x]+cost[i]<dis[u]){
dis[u]=dis[x]+cost[i];
pre[u]=i;
incf[u]=min(incf[x],val[i]);
if(!vis[u]) vis[u]=1,Q.push(u);
}
}
vis[x]=0;
}
if(dis[T]==inf) return false;
return true;
}
void update(){
int x=T;
while(x!=S){
int i=pre[x];
val[i]-=incf[T];
val[i^1]+=incf[T];
x=to[i^1];
}
ans+=dis[T]*incf[T];
}
int main(){
n=rd();S=0,T=n+1;
for(register int i=1;i<=n;i++) a[i]=rd(),sum+=a[i];
sum/=n;
for(register int i=1;i<=n;i++){
a[i]-=sum;
if(a[i]>0) add(S,i,a[i],0);
else add(i,T,-a[i],0);
}
for(register int i=2;i<n;i++)
add(i,i-1,inf,1),add(i,i+1,inf,1);
add(1,2,inf,1),add(1,n,inf,1);
add(n,n-1,inf,1),add(n,1,inf,1);
while(spfa()) update();cout<<ans<<endl;
return 0;
}