AcWing 122. 糖果传递【贪心】【《算法竞赛进阶指南》,微软面试题 , HAOI2008】

Derrickhang

已于 2022-03-27 15:21:24 修改

阅读量327

点赞数 1

文章标签： c++ 数据结构算法《算法竞赛进阶指南》

于 2022-03-26 19:05:38 首次发布

本文链接：https://blog.csdn.net/Derrickhang/article/details/123761077

版权

AcWing 122. 糖果传递

一、题目链接
二、题目分析
- （一）算法标签
- （二）解题思路
三、AC代码
四、其它题解

一、题目链接

AcWing 122. 糖果传递
 进阶题目 AcWing 105. 七夕祭

二、题目分析

（一）算法标签

贪心排序中位数推公式

（二）解题思路

设 $x_i$ 为i号小朋友(初始有 $a_i$ 个糖果)向i-1号小朋友传递的糖果数(其中，i=2,3,4,…,n, 若 $x_i$ 为负数，则表示i-1号小朋友向i号小朋友传递的糖果数)，特别地当i=1时， $x_1$ 表示1号小朋友向n号小朋友传递的糖果数

目标： $\vert x_1 \vert + \vert x_2\vert+\vert x_3 \vert+ \dots +\vert x_{n-1} \vert+\vert x_n \vert$ 最小
由题意，可得：
$\begin{cases} x_1=x_2+a_1-\bar a \\ x_2=x_3+a_2-\bar a \\ \cdots \\ x_{n-1}=x_n+a_{n-1}-\bar a \\ x_n=x_1+a_n-\bar a \\ \end{cases}$

n个方程相加得到一个恒等式： $0=a_1+a_2+\cdots+a_n-n\bar a$ ，所以上面的方程组有n-1个有效方程，n个未知元，那么可以用其中一个未知元来表示其它n-1个未知元

对上述方程组作一下变换，可以得到：
$\begin{cases} x_1=x_1-0 \\ x_2=x_3-[(n-1)\bar a - a_n-a_{n-1}-\cdots-a_2]\\ \cdots \\ x_{n-1}=x_1-(2\bar a-a_n-a_{n-1}) \\ x_n=x_1-(\bar a - a_n) \\ \end{cases}$
等价于下面方程组：
$\begin{cases} x_1=x_1-c_1 \\ x_2=x_3-c_2\\ \cdots \\ x_{n-1}=x_1-(2\bar a-a_n-a_{n-1}) \\ x_n=x_1-(\bar a - a_n) \\ \end{cases}$

其中，
$\begin{cases} c_1=0 \\ c_2=(n-1)\bar a - a_n-a_{n-1}-\cdots-a_2 \\ \cdots \\ c_{n-1}=2\bar a-a_n-a_{n-1}\\ c_n=\bar a - a_n\\ \end{cases}$
可以发现当 $i > 1$ 时 $c_i=c_{i+1}+\bar a -a_i$
回到目标：
此时，
$\vert x_1 \vert + \vert x_2\vert+\vert x_3 \vert+ \dots +\vert x_{n-1} \vert+\vert x_n \vert$ 最小
等价于 $\vert x_1-c_1 \vert + \vert x_1-c_2\vert+\vert x_1-c_3 \vert+ \dots +\vert x_1-c_{n-1} \vert+\vert x_1-c_n \vert$ 最小 (其中 $c_1,c_2,\cdots,c_n$ 均为常数)
当 $x_1$ 取 $c_1,c_2,\cdots,c_n$ 的中位数即可使上述式子的值最小
此时，问题转换成了AcWing 104. 货仓选址【贪心】【《算法竞赛进阶指南》】

三、AC代码

解法一:

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 1e6 + 10;

int a[N], c[N];
int n;


int main()
{
    cin >> n;
    LL sum = 0;
    for (int i = 1; i <= n; i ++ )
    {
        scanf("%d", &a[i]);
        sum += a[i];
    }
    // 每个小朋友最后的糖果数一样，所以平均数一定是整数
    LL avg = sum / n;
    
    // 求c[i]
    for (int i = n; i > 1; i -- )
        c[i] = c[i + 1] + avg - a[i];
    
    // 找c[i]的中位数
    sort(c + 1, c + 1 + n);
    LL mid = c[1 + n >> 1];
    
    // 求结果
    LL res = 0;
    for (int i = 1; i <= n; i ++ )
        res += abs(c[i] - mid);
    
    cout << res << endl;
    return 0;
}