糖果传递 ——《算法竞赛进阶指南》

糖果传递

题目：

题目描述：

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

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

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

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

输入格式

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

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

输出格式

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

数据范围

1≤n≤1000000,
0≤a[i]≤2×10^9,
数据保证一定有解。

输入样例：

4
1
2
5
4

输出样例：

4

时间复杂度：O（nlog2n）

算法: 贪心、中位数、推公式

思路分析：

核心： 经过一系列的数学求解操作，转化成求中位数的问题。

首先我们可以把题目中的内容抽象出来这么一个模型，ai代表是第i个小朋友最开始拥有的糖果树，xi表示ai小朋友给ai-1小朋友的糖果数量（当i等于1时，x1表示a1小朋友给an小朋友的糖果数量），如果xi是整数就表示ai小朋友给了ai-1小朋友xi个糖果，如果xi是负数，可以看作是ai-1小朋友给了ai小朋友xi个糖果。题目让我们最终当每个小朋友手里的糖果数均等时求ans = |x1| + |x2| + |x3| + …… |xn|的最小值。

根据上图我们可以得到方程组：

每个方程表示第i个同学拥有的糖果数ai在给了第i - 1个朋友xi个糖果和得到第i + 1个小朋友xi + 1个糖果后剩余的糖果数(值等于最终结果拥有的糖果数量，即糖果总数的平均数)。

从图中可以不难看出，该方程组中共有n个方程以及n个未知量，但从前n - 1个方程可以推出最后一个方程，所以其实是由n - 1 个独立方程和n个未知量，根据线性代数的知识，R（A | b）= R(A) = n - 1 < n,所以该方程的自由变元数为1，我们对于任意的xi都可以通过xj去表示，虽然我们无法求得该方程的唯一解，但是可以对xi通过xj表示，将其变成单变量的最大最小值问题。这里我们先让x1为自由变元去表示任意的xi（即xj = x1）：

接下去我们可以将x1， x2 …… xn都带入到答案方程 ans = |x1| + |x2| + |x3| + …… |xn|中去：

|x1 - ci|的几何意义表示数轴上x1到ci的距离，因此本题的问题经过一系列数学知识后，转变成了：给定数轴上n个点c1 ~ cn，然你找出一个点x1，使得其到c1 到 cn的距离之和最小， 而这个点就是c1到cn这些数的中位数。（原因在这里不细证明，可以看下绝对值不等式的一道模板题来进一步学习 货仓选址）

代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
const int N = 1e6 + 5;
int n; //n:小朋友个数   
int a[N];   //a数组：每个小朋友初始状态拥有糖果数
ll ave, sum, c[N];   //ave：糖果总数的平均数   sum：糖果总数
int main()
{
    cin>>n;
    //输入并求总糖果数
    for(int i = 1; i <= n; i ++){
        cin>>a[i];
        sum += a[i];
    }

    ave = sum / n;  //求一下糖果总数的平均数

    //通过递推求出c数组的值
    for(int i = n; i >= 2; i --)c[i] = c[i + 1] + ave - a[i];
    c[1] = 0;

    sort(c + 1, c + n + 1); //对c数组从小到大排序，方便后序求c数组的中位数

    ll ans = 0;
    for(int i = 1; i <= n; i ++)ans += abs(c[i] - c[(1 + n) / 2]);  //通过中位数求得最小代价和
    cout<<ans<<'\n';

    return 0;
}