Alibaba在线编程原题

本题是参加Alibaba实习，第一个环节在线编程题原题，不知道自己写的对不对，大家帮忙看看，希望不涉及版权问题。

------------------------------------题目-------------------------------------

有n个打包机器从左到右一字排开，上方有一个自动装置会抓取一批物品到每个打包机上，放到机每个器上的这些物品数量有多有少，由于物品数量不相同，需要工人将每个机器上的物品进行移动从而到达物品数量相等才能打包。每个物品重量太大，每次只能搬一个物品进行移动；为了省力，只能在相邻的机器上移动。请计算在搬动最小次数的前提下，使每个机器上的物品数量相等。如果不能使每个机器上的物品相同，返回-1。

 例如
Input: [1,0,5]  

Output: 3

 [1,0,5] 表示有3个机器，每个机器上分别有1、0、5个物品，
 经过这些移动后：
  1st:    1     0 <-- 5    =>    1     1     4
  2nd:    1 <-- 1 <-- 4    =>    2     1     3    
  3rd:    2     1 <-- 3    =>    2     2     2  

  移动了3次，每个机器上的物品相等。

再比如
Input: [0,3,0]
Output: 2

[0,3,0] 表示有3个机器，每个机器上分别有0、3、0个物品，
经过这些移动后：
1st:    0 <-- 3    0   =>    1     2    0
2nd:    1   2   -> 0  =>    1     1    1
移动了2次，每个机器上的物品相等。

再比如
Input: [2,2,3]
Output: -1
[2,2,3] 表示有3个机器，每个机器上分别有2、2、3个物品

-----------------------------------------代码--------------------------------------

import java.util.*;
public class Main
{
/** 请完成下面这个函数，实现题目要求的功能 **/
/** 当然，你也可以不按照这个模板来作答，完全按照自己的想法来 ^-^ **/
static int solution(int[] machines)
{
int total = 0;    //记录总的物体个数
int aver;    //最后平均每个机器上的物体个数

int n = 0;    //记录需要的步数

for (int i = 0; i < machines.length; i++)//实现求总的物体个数
{
total += machines[i];
}
if (total % machines.length != 0)    //如果不能整除，说明无法达到题目要求即每个机器上物体个数相同
return -1;

aver = total / machines.length;    //求出平均每个机器上的物体个数

for (int i = 0; i < machines.length - 1; i++)    //对前n-1个机器上的物体进行处理
{
if (machines[i] == aver)    //如果第 i 个机器上的物体数目等于平均数，即本机器可以不做处理
continue;
                        while (machines[i] > aver)     //如果第 i 个机器上的物体数目大于平均数
{
machines[i]--;
machines[i + 1]++;
n++;
}
while (machines[i] < aver)        //如果第 i 个机器上的物体数目小于平均数，
{
if(machines[i]<0)
machines[i] = 0;
machines[i]++;
machines[i + 1]--;
n++;
}
}
return n;
}
public static void main(String[] args)    //主函数
{
Scanner in = new Scanner(System.in);
int res;
String[] machinesTmp = in.nextLine().trim().split(",");
int[] machines = new int[machinesTmp.length];
for (int i = 0; i < machines.length; i++)
{
machines[i] = Integer.parseInt(machinesTmp[i]);
}


res = solution(machines);
System.out.println(String.valueOf(res));


}
}