HDOJ1326 Box of Bricks

#include<stdio.h>#include<stdlib.h>using namespace std;#define NUM 50 +5int main(){    int n,total,average,step,hgt[NUM];    int i,count=1;    while(scanf("%d",&n),n){          memset(hgt,0,sizeof(hgt));          total=0;          average=0;          step=0;          for(i=0;i<n;i++){                  scanf("%d",&hgt[i]);                  total=total+hgt[i];          }          average=total/n;          for(i=0;i<n;i++)                  if(hgt[i]>average)                      step=step+hgt[i]-average;                                printf("Set #%d\nThe minimum number of moves is %d.\n\n",count,step);          count++;              }  }