SDJZ 摆积木

题目描述

小明很喜欢玩积木。一天，他把许多积木块组成了好多高度不同的堆，每一堆都是一个摞一个的形式。然而此时，他又想把这些积木堆变成高度相同的。但是他很懒，他想移动最少的积木块来实现这一目标，你能帮助他吗？

输入格式

输入包含多组测试样例。每组测试样例包含一个正整数n，表示小明已经堆好的积木堆的个数。
接着下一行是n个正整数，表示每一个积木堆的高度h，每块积木高度为1。其中1<=n<=50,1<=h<=100。
测试数据保证积木总数能被积木堆数整除。
当n=0时，输入结束。

输出

对于每一组数据，输出将积木堆变成相同高度需要移动的最少积木块的数量。
在每组输出结果之间输出一个空行。

样例输入

6
5 2 4 1 7 5
0

样例输出

5

 

代码：

#include <iostream>
using namespace std;
int main()
{
  int n,p[10000],f[10000];
  int sum;
  int average;
  while(cin>>n)
  {
    int sum2=0;
    sum=0;
      if(n==0)
      {
          break;
      }
    for(int i=0;i<n;i++)
    {
      cin>>p[i];
      sum+=p[i];
    }
    average=sum/n;
    for(int j=0;j<n;j++)
    {
        f[j]=p[j]-average;
        if(f[j]>0)
        {
            sum2+=f[j];
        }
    }
    cout<<sum2<<endl;
    cout<<endl;
  }

  return 0;
}