均分糖果——2021年2-7更

最新推荐文章于 2024-09-20 13:11:30 发布

原创最新推荐文章于 2024-09-20 13:11:30 发布 · 3.1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #贪心算法

这篇博客介绍了如何使用贪心算法解决一个关于小朋友糖果分配的问题。当n位小朋友围坐一圈，初始糖果数量不同时，通过计算平均值并进行传递，可以找到最少传递次数使所有人糖果数相同。博主分享了错误的暴力枚举思路和正确的贪心算法实现，包括代码细节，最终得出最少传递的糖果数。

    题目
  小A班的 n 位小朋友在操场围坐成一圈，每人手里有数量不等的糖果，小朋友们可以向左右传递，请问至少需要传递多少颗糖果才能使得所有小朋友手里的糖果数一样。

输入
第一行输入一个正整数n，表示小朋友的个数。（1≤n≤1000000 ）

接下来n行，每行一个整数，表示每个小朋友最开始自己手里的糖果数。
输出
一个整数，表示最少传递的糖果数。（数据保证有解）
输入样例
4
1
2
5
4
输出样例
4

思路：

原本以为是个水题，然后没看数据点就直接暴力枚举，然后就在这里插入图片描述

正解：

~~打表AC~~ 贪心思想,首先求出总数/n(个数)，在利用类似前缀和思想将结果存在一个数组里，排序后求出中值，在用每个数和中值相减即可。

代码

#include<bits/stdc++.h>
using namespace std;
long long a[1000100],x[1000100];
long long mid,ans,n,sum;
int main()
{
	ios::sync_with_stdio(false);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		sum+=a[i];
	}
	sum/=n;
	for(int i=1;i<=n;i++)
		x[i]=x[i-1]-a[i]+sum;
	sort(x+1,x+1+n);
	mid=x[(n+1)/2];
	for(int i=1;i<=n;i++)
		ans+=abs(x[i]-mid);
	cout<<ans;
	return 0;
}