糖果传递题解

糖果传递题解

我们不妨设第n个人给第一个人的糖果数为x1,第一个人给第二个人的为x2,第二个给第三个的为x3，......,第n-1个人给第n个人的为xn,则：

答案为\(|x_{1}|+|x_{2}|+|x_{3}|+......+|x_{n}|\)

且满足条件：

\(a_{i}+x_{i}-x_{i+1}=average\),

则：\(x_{i+1}=a_{i}-average+x_{i}\)

\(x_{2}=a_{1}-ave+x_{1};\)

\(x_{3}=a_{2}-ave+x_{2}=(a_{1}-ave)+(a_{2}-ave)+x_{1}\)

\(x_{4}=a_{3}-ave+x_{3}=(a_{1}-ave)+(a_{2}-ave)+(a_{3}-ave)+x_{1}\)

......

\(x_{n}=(a_{1}-ave)+......+(a_{n}-1-ave)+x_{1};\)

设\(c_{n}=\sum _{i=1}^{n}ave-a_{i}\)

答案为：\(|x_{1}|+|x_{1}+c_{1}|+......+|x_{1}+c_{n}-1|\)

据初中知识可知：我们必选中位数。

下面是代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1000006;
ll n,a[N],ave,c[N],sum=0,ans=0;
inline int read(){
   int T=0,F=1; char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-') F=-1; ch=getchar();}
   while(ch>='0'&&ch<='9') T=(T<<3)+(T<<1)+(ch-48),ch=getchar();
   return F*T;
}
int main(){
    n=read();
    for(int i=1;i<=n;++i) a[i]=read(),sum+=a[i];
    ave=sum/n,c[0]=0;
    for(int i=1;i<=n;++i) c[i]=c[i-1]-a[i]+ave;
    sort(c,c+n),ave=c[n/2];
    for(int i=0;i<n;++i) ans+=abs((ave-c[i]));
    printf("%lld\n",ans);
    return 0;
}

转载于:https://www.cnblogs.com/ljk123-de-bo-ke/p/11318189.html