BZOJ-1045-[HAOI2008] 糖果传递（中位数原理）

Posted 2020-10-11 I__am

Description

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

Input

第一行一个正整数nn<=1‘000‘000，表示小朋友的个数．

接下来n行，每行一个整数ai，表示第i个小朋友得到的糖果的颗数．

Output

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

Sample Input

4
1
2
5
4

Sample Output

4

 

题解

这道题对于最终小朋友手中的糖的数量我们是可以算出来的，我们用ave来表示

我们假设G_i表示第i个人给第i-1个人糖的数量，G₁表示第1个人给第n个人

那么最后答案就是|G₁|+|G₂|+···+|G_n|

那么到最后

第一个人的糖就是A₁-G₁+G₂=ave

第二个人的糖就是A₂-G₂+G₃=ave

······

第n个人的糖就是A_n-G_n+G₁=ave

这里我们假设C_i=C_i-1+A_i-ave

所以通过第一个人的糖，我们可以推出G₂=G₁-C₁

通过第二个人，可以推出G₃=G₁-C₂

······

第n个人，可以推出G_n=G₁-C_n-1

所以最后答案就变成了|G₁|+|G₁-C₁|+|G₁-C₂|+···+|G₁-C_n-1|

对于求最后答案，问题就变成了给你坐标轴上的n个点，要你找到一个点，使得这个点到所有的点的距离之和最小

而这个点的坐标就是坐标轴上点的中位数

 1 #include<bits/stdc++.h>
 2 #define N 1000005
 3 #define ll long long
 4 using namespace std;
 5 int n,ave;
 6 ll sum,ans;
 7 int a[N],c[N];
 8 int main(){
 9     scanf("%d",&n);
10     for (int i=1;i<=n;i++){
11         scanf("%d",&a[i]);
12         sum+=a[i];
13     }
14     ave=sum/n;
15     for (int i=2;i<=n;i++)
16         c[i]=c[i-1]+a[i]-ave;
17     sort(c+1,c+1+n);
18     int mid=c[(n>>1)+1];
19     for (int i=1;i<=n;i++)
20         ans+=abs(mid-c[i]);
21     printf("%lld\n",ans);
22     return 0;
23 }

View Code