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

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

输入
第一行输入一个正整数n,表示小朋友的个数。(1≤n≤1000000 )

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

思路:

原本以为是个水题,然后没看数据点就直接暴力枚举,然后就均分糖果——2021年2-7更_第1张图片

正解:

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

代码

#include
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;
}

本蒟蒻第一次写题解,求大佬们多多关照

你可能感兴趣的:(c++,贪心算法)