1. 程式人生 > >Alibaba線上程式設計原題

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));


}
}