ACcoders Problem 2042 题解

Tobbichi_Origami

已于 2023-02-15 16:28:14 修改

阅读量91

点赞数 1

分类专栏：一本通提高篇补全计划文章标签：算法

于 2023-02-15 16:25:56 首次发布

本文链接：https://blog.csdn.net/gyk038/article/details/129045119

版权

一本通提高篇补全计划专栏收录该内容

28 篇文章 0 订阅

订阅专栏

文章描述了一个数学问题，涉及n个小朋友和他们的糖果分配。每个小朋友有特定数量的糖果，目标是通过最小代价的传递使得所有小朋友的糖果数量相等。解决方案涉及到计算平均数，建立传递模型，并通过寻找中位数来确定最小代价。最后给出了C++代码实现。

摘要由CSDN通过智能技术生成

题意

有 $n$ 个小朋友，他们围成了一个圈，每个人有 $a_{i}$ 个糖果。每人每次传递一个糖果代价为 $1$ 。求使所有小朋友获得均等糖果的最小代价。

思路

我们设 $x_{i}$ 为第 $i$ 个小朋友传给第 $i + 1$ 个小朋友的糖果数量， $a v e$ 为所有小朋友的糖果的平均数。则有：
$ave=\frac{1}{n}(a_{1}+a_{2}+…a_{n})$

$a_{i}-x_{i}+x_{i-1}=ave$
我们再将第二个式子展开，即：
$a_{1}-x_{1}+x_{n}=ave$

$a_{2}-x_{2}+x_{1}=ave$

$\dots$

$a_{n}-x_{n}+x_{n-1}=ave$
然后再移项
$x_{1}=x_{n}-(ave-a_{1})$

$x_{2}=x_{n}-(ave-a_{2})$

$\dots$

$x_{n}=x_{n-1}-(ave-a_{n})$
我们再将其逐个由之前的式子替换
$x_{1}=x_{n}-(ave-a_{1})$

$x_{2}=x_{n}-(2\times ave-a_{1}-a_{2})$

$\dots$

$x_{n}=x_{n}-(n×ave−a_{1}−a_{2}−…−a_{n})$
因为我们要求的是传递最小的代价，即为：
$x_{1}|+|x_{2}|+…+|x_{n}|$
我们发现，可以将原来的式子带入其中，并且此式后面的常数部分我们可以整理一下用 $c_{i}$ 代替。
$|x_{n}-(ave-a_{1})|+|x_{n}-(2\times ave-a_{1}-a_{2})|+…+|x_{n}-(n×ave−a_{1}−a_{2}−…−a_{n})|$

$x_{n}-c_{1}|+|x_{n}-c_{2}|+…+|x_{n}-c_{n}|$
现在，我们就可以将其转化为一个数轴上求距离的集合问题。我们再将 $x_{n}-c_{1}|$ 和 $x_{n}-c_{n-1}|$ 分为一组，随后以此类推，发现 $x_{n}$ 的取值范围可以用下图来表示（最外层就是 $x_{n}-c_{1}|$ 和 $x_{n}-c_{n-1}|$ ，再向里层就以此类推），发现所有组别的取值范围的交集为中心蓝色的部分，即所有 $c_{i}$ 的中位数。
在这里插入图片描述
代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
int a[1000001];
int c[1000001];
int n,ave;
signed main()
{
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		ave+=a[i];
	}
	ave/=n;
	for(int i=1;i<=n;i++)
		c[i]=c[i-1]+ave-a[i-1];//等价
	sort(c+1,c+n+1);
	int ans=0,mid=c[(n+1)/2];
	for(int i=1;i<=n;i++)
		ans+=abs(mid-c[i]);
	printf("%lld",ans);
	return 0;
}