有n个人围圆桌而坐,每个人有Ai个金币,可以给左右相邻的人一些金币,使得最终所有人金币数相等,求最小金币转移数。
思路:
①所有金币数/人数=m(每个人最终应有的金币数)
②对于任意两个人a,b,a给b n1个金币,b给a n2个金币,则a与b之间关系为(n1-n2)(可正可负);
③对于任意第i个人存在两种关系,给金币给i-1,从i+1获得金币。A[i]-x[i]+x[i+1]=m;
④用x1将其他x表示:
x2=x1-A[i]+m;
x3=x2-A2+m=x1-A[1]-A[2]+2m;
x4=x3-A3+m=x1-A[1]-A[2]-A[3]+3m;
···
⑤为了简化,用C[1]=A[1]-m,C[2]=c[1]+A[2]-m ··· ···
则④可表示为:
x2=x1-C[1];
x3=x1-C[2];
x4=x1-C[3];
···
⑥最终结果使得|x1|+|x1-C1|+|x1-C2|+···+|x1-Cn-1|最小。即找到x1,使得到所有C[i]点的距离之和最小。
所以C的中位数为最优解。
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 5;
ll A[maxn];
ll C[maxn];
ll n;
ll sum;
ll m;
int main()
{
ios::sync_with_stdio(false);
while (cin >> n)
{
memset(A, 0, sizeof(A));
memset(C, 0, sizeof(C));
sum = 0;
for (int i = 1; i <= n; i++)
{
cin >> A[i];
sum += A[i];
}
m = sum / n;
for (int i = 1; i < n; i++)
{
C[i] = C[i - 1] + A[i] - m;
}
sort(C, C + n);
ll x1 = C[n / 2], ans = 0;
for (int i = 0; i < n; i++)
{
ans += abs(x1 - C[i]);
}
cout << ans << endl;
}
`` return 0;
}