初级算法：小明移动积木

这是我面试遇到的三道题之一，这种题真是坑人，很简单，但就是读不懂题目，无从下手，数学没学好。
题目描述
小明很喜欢玩积木。一天，他把许多积木块组成了好多高度不同的堆，每一堆都是一个摞一个的形式。然而此时，他又想把这些积木堆变成高度相同的。但是他很懒，他想移动最少的积木块来实现这一目标，你能帮助他吗？
输入格式
输入包含多组测试样例。每组测试样例包含一个正整数n，表示小明已经堆好的积木堆的个数。
接着下一行是n个正整数，表示每一个积木堆的高度h，每块积木高度为1。其中1<=n<=50,1<=h<=100。
测试数据保证积木总数能被积木堆数整除。
当n=0时，输入结束。

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

样例输入
6
5 2 4 1 7 5
0

样例输出
5

public class Main {
public static void main(String[] args) {
	Scanner scanner = new Scanner(System.in);
	while (scanner.hasNextInt()) {
		int n = scanner.nextInt();
		if (n == 0) {
			scanner.close();
			System.exit(0);
		}
		int[] arr = new int[n];
		int sum = 0, avg = 0, step = 0;
		for (int i = 0; i < arr.length; i++) {
			arr[i] = scanner.nextInt();
		}
		for (int i = 0; i < arr.length; i++) {
			sum += arr[i];
		}
		avg = sum / n;
		if (sum % n != 0) {
			System.out.println("data error");
		}
		for (int i = 0; i < arr.length; i++) {
			if (arr[i] > avg) {
				step = step + (arr[i] - avg);//没有想到这么简单，我面试题写了一大堆。
			}
		}
		System.out.println(step);
	}
}

}