贪心-acwing-122-糖果传递

贪心-acwing-122-糖果传递

题目：
有n个小朋友坐成一圈，每人有a[i]个糖果。

每人只能给左右两人传递糖果。

每人每次传递一个糖果代价为1。

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

输入格式
第一行输入一个正整数n，表示小朋友的个数。

接下来n行，每行一个整数a[i]，表示第i个小朋友初始得到的糖果的颗数。

输出格式
输出一个整数，表示最小代价。

数据范围
1≤n≤1000000
数据保证一定有解。

输入样例：
4
1
2
5
4
输出样例：
4

题解：

$如上图，我们设相邻的小朋友给第i个小朋友x_i个糖果(x_i>0表示给出，x_i<0表示得到)。$

$$\begin{gathered} 即： \\ {a}_1+x_1-x_n=\bar{a},① \\ {a}_2+x_2-x1=\bar{a},② \\ ...... \\ {a}_{n-1}+x_{n-1}-x_{n-2}=\bar{a},③ \\ {a}_n+x_n-x_{n-1}=\bar{a}。④ \end{gathered}$$

$其中\bar{a}=\frac{{\sum }_{i=1}^n{a}_i}{n}。$

$求{\sum }_{i=1}^n|x_i|的最小值。$

$因为a_i是常数，我们试着用a_i来表示x_i。$

$$\begin{gathered} 上述①式得：x_1=x_n+\bar{a}-a_1,⑤ \\ 将⑤代入②得：x_2=x_n+2\bar{a}-a_1-a_2,⑥ \end{gathered}$$
$依次类推x_i=x_n+i\bar{a}-a_1-a_2-...-a_i,(i>=1)。$

$$\begin{gathered} 由于a_i是常数，故设x_i=x_n-C_i，其中C_i=i\bar{a}-{\sum }_{k=1}^i{a}_k。 \\ 则C_i=C_{i-1}+\bar{a}-a_i，C_0=0,(i>=1)。 \end{gathered}$$

$现求{\sum }_{i=1}^n|x_i|即求{\sum }_{i=1}^n|x_n-C_i|即选择一个点x_n到数轴上n个点C_1,C_2,...,C_n距离和。$

$这样就转化到选址问题上来了——$《货舱选址》。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define inf 0x7fffffff
using namespace std;
const int N=1e6+10;
int a[N],n;
ll C[N];
ll ave,ans;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        ave+=a[i];
    }

    ave/=n;

    for(int i=1;i<=n;i++)
        C[i]=C[i-1]+ave-a[i];

    sort(C+1,C+n+1);

    for(int i=1;i<=n/2;i++)
        ans+=(C[n-i+1]-C[i]);

    printf("%lld\n",ans);

    return 0;
}