当前位置: 代码迷 >> 综合 >> [bzoj1045][bzoj3293][数论]糖果传递/分金币
  详细解决方案

[bzoj1045][bzoj3293][数论]糖果传递/分金币

热度:28   发布时间:2023-12-19 05:26:11.0

Description

有n个小朋友坐成一圈,每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为1。

Input

第一行一个正整数nn<=1’000’000,表示小朋友的个数. 接下来n行,每行一个整数ai,表示第i个小朋友得到的糖果的颗数.

Output

求使所有人获得均等糖果的最小代价。

Sample Input

4

1

2

5

4

Sample Output

4

题解

设a[i]表示第i个人给了第i+1个人多少
设s[i]表示第i个人原本有多少个
容易看出最后每个人的糖果数都是平均数
可列方程
s[1]+a[n]-a[1]=avg
s[2]+a[1]-a[2]=avg
….
s[n]+a[n-1]-a[n]=avg
移项可得
a[1]=s[1]+a[n]-avg
a[2]=s[2]+a[1]-avg=s[2]+s[1]+a[n]-2*avg
a[3]=s[3]+a[2]-avg=s[3]+s[2]+s[1]+a[n]-3*avg

把a[n]延后,可发现前面项均为前缀和减去平均数形式,设他们为C[i]
问题可变为求sigma(a[n]-C[i])
C数组为定值,原问题仍可转化为在数轴上有n个点,选一个点使得这n个点到这个点的总距离最小
容易看出这个点是中位数

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
int n;
LL a[1110000],avg;
int main()
{scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%lld",&a[i]),avg+=a[i];avg/=n;for(int i=1;i<=n;i++)a[i]=a[i-1]+a[i]-avg;sort(a+1,a+1+n);int tmp=n/2+1;LL ans=0;for(int i=1;i<=n;i++)ans+=abs(a[tmp]-a[i]);printf("%lld\n",ans);return 0;
}