题目描述:lrj厚白书第一章第三个例题
题目分析:首先来模拟spreading。对于i,他只可能从他的左边得到xi-1枚金币,从他的右边得到xi+1枚金币,而最终的i只能拥有平均数的金币。将这写等式列出,就可以得到一个含有绝对值的多项式。这样就可以利用绝对值的集合意义得到这个多项式的取最小值的情况。
注意:求c[i]时用递推,否则将会tle。
下面是代码:
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
using namespace std;
const int maxn = 1000005;
int n;
int w[maxn];
int c[maxn];
int main()
{
while(~scanf("%d",&n)) {
long long tot = 0;
for(int i = 0; i < n ;i++) {
scanf("%d",&w[i]);
tot += w[i];
}
int ave = tot/n;
c[0] = 0;
for(int i = 1; i < n; i++) {
c[i] = c[i-1] + ave - w[i];
}
sort(c,c+n);
int t = c[n/2];
long long ans = 0;
for(int i = 0; i < n; i++) {
ans += abs(t - c[i]);
}
printf("%lld\n",ans);
}
return 0;
}