[网络流24题] 负载平衡

741. [网络流24题] 负载平衡

★★   输入文件：overload.in   输出文件：overload.out   简单对比
时间限制：1 s   内存限制：128 MB

«问题描述：
G 公司有n 个沿铁路运输线环形排列的仓库，每个仓库存储的货物数量不等。如何用最
少搬运量可以使n 个仓库的库存数量相同。搬运货物时，只能在相邻的仓库之间搬运。
«编程任务：
对于给定的n 个环形排列的仓库的库存量，编程计算使n 个仓库的库存数量相同的最少
搬运量。
«数据输入：
由文件overload.in提供输入数据。文件的第1 行中有1 个正整数n（n<=100），表示有n
个仓库。第2 行中有n个正整数，表示n个仓库的库存量。
«结果输出:
程序运行结束时，将计算出的最少搬运量输出到文件overload.out中。
输入文件示例 输出文件示例
overload.in
5

17 9 14 16 4

overload.out

11

/*显示代码纯文本

提供最小费用流思路：
相邻的仓库之间连双向边，容量无穷大，费用为1。
如果仓库i中的货物小于平均值，则从i向汇点连边（表示i需要接收货物），容量为average - A[i]。
如果仓库i中的货物大于平均值，则从源点向i连边（表示i可以输出货物），容量为A[i] - average。
 
本代码思路：套用DP（平衡序列） 
*/
#include<cstdio>
#include<algorithm>
const int N=205;
int n,tot,a[N],b[N];
int main(){
	freopen("overload.in","r",stdin);
	freopen("overload.out","w",stdout);
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),tot+=a[i];
	tot/=n;
	for(int i=1;i<=n;i++){
		b[i]=tot-a[i];
		b[i]+=b[i-1];
	} 
	std::sort(b+1,b+n+1);
	int ans=0;
	for(int i=1;i<=n;i++) ans+=abs(b[n+1>>1]-b[i]);
	printf("%d",ans);
	return 0;
}