题意
有n个小朋友坐成一圈，每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为1。
思路
假设平均数是x，且a1给an了k个（k<0说明是an给a1了-k个），那么总代价就可以算出来：
| an |
|
|
|
| an+k |
a1-k |
a2 |
|
| 代价：|k| |
x |
a1+a2-x-k |
a3 |
|
代价：|a1-x-k| |
x |
a1+a2+a3-2x-k |
|
|
代价：|a1+a2-2x-k| |
x |
|
|
|
代价：|a1+a2+a3-3x-k| |
令bi=sum(a1..i)-ix，则总代价=
sum|bi-k|。易知k为中位数时此值最小。问题转化为求中位数……
用自己快速选择第K大模板TLE……sort直接过了……
代码
[cpp]
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#define MID(x,y) ((x+y)/2)
#define MEM(a,b) memset(a,b,sizeof(a))
#define REP(i, begin, end) for (int i = begin; i <= end; i ++)
using namespace std;
const int maxn = 1000005;
int a[maxn];
long long sum[maxn], b[maxn];
int main(){
int n;
scanf("%d", &n);
sum[0] = 0;
REP(i, 1, n){
scanf("%d", &a[i]);
sum[i] = sum[i-1] + a[i];
}
long long average = sum[n] / n;
REP(i, 0, n-1){
b[i] = sum[i] - (i) * average;
}
sort(b, b+n);
long long res = 0;
REP(i, 0, n-1){
res += abs(b[i] - b[n/2]);
}
printf("%lld
", res);
return 0;
}
[/cpp]