acwing122.糖果传递

骰子.w

已于 2023-10-24 15:32:34 修改

阅读量105

点赞数

分类专栏： acwing算法竞赛进阶指南文章标签： 1024程序员节算法

于 2023-10-24 15:18:36 首次发布

本文链接：https://blog.csdn.net/weixin_73355895/article/details/134012071

版权

acwing算法竞赛进阶指南专栏收录该内容

7 篇文章 0 订阅

订阅专栏

传送门：122. 糖果传递 - AcWing题库

此题题解是根据b站董晓老师的视频所写，可以自行去b站搜索董晓老师的视频讲解（老师的视频非常详细易懂）

数学归纳总结：

假设a[i]表示初始每个人糖果数量，x[i]表示第i个人向左传递的糖果数量，ave表示平均每个人应得的数量。

题目要求就相当于是最小化| x[1] | + | x[2] | +......+ | x[n] | ;

那么：

ave=a[i]-x[i]+x[i+1](i<n)

ave=a[i]-x[i]+x[1] (i==n)

即：

ave=a[1]-x[1]+x[2];

ave=a[2]-x[2]+x[3];

......

(用x[1]表示)可转化为：

x[2]=ave+x[1]-a[1];

x[3]=ave+x[2]-a[2];

x[4]=ave+x[3]-a[3];

......

x[n]=ave+x[n-1]-a[n-1];

全部用x[1]表示为：

x[2]=ave+x[1]-a[1];

x[3]=x[1]+ave-a[1]+ave-a[2](ave+ave+x[1]-a[1]-a[2]);

x[4]=x[1]+ave-a[1]+ave-a[2]+ave-a[3];

......

x[n]=x[1]+ $\sum_{i=1}^{n-1}(ave-a[i])$

令c[i]= $\sum_{j=1}^{i-1}(a[j]-ave)$

那么x[i]=x[1]-c[i];

题目所求为：|x[1]|+|x[1]-c[2]|+......+|x[1]-c[n]|;

要使得其最小，令x[1]取c的中位数即可

（取中位数的原因可以做做另一道题：104. 货仓选址 - AcWing题库

此处贴出蓝书对其（货仓选址）关于选取中位数的证明：把A[1]~A[N] 排序，设货仓建在 X 坐标处，X 左侧的商店有P家，右侧的商店有Q家。若 P<Q，则每把货仓的选址向右移动1单位距离，距离之和就会变小 Q-P。同理，若 P>Q，则货仓的选址向左移动会使距离之和变小。当 P=Q 时为最优解。）

#define _CRT_SECURE_NO_WARNINGS
//#pragma GCC optimize(1)
//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
using namespace std;
#define N 5000005 
#define M 1331 
#define PII pair<int,int>
#define PIII pair<pair<int,int>,int>  
#define VPII vector<PII>
#define VPIII vector<PIII>
#define inf 0x3f3f3f3f3f3f  
#define ll long long
#define ull unsigned long long
const double eps = 1e-6;
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define mod 1000000007
ll a[N],x[N],c[N];
ll n,m;
void solve() {
	cin >> n;
	ll ave = 0;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		ave += a[i];
	}
	ave /= n;
	for (int i = 2; i <= n; i++) {
		x[i] =  a[i]-ave+x[i-1];
	}
	sort(x + 1, x + 1 + n);
	m = n / 2;
	if (n & 1) m++;
	ll ans = 0;
	for (int i = 1; i <= n; i++) {
		ans += labs(x[m] - x[i]);
	}
	cout << ans << "\n";
}
int main()
{
	IOS;
	solve();
	return 0;
}