题目:
假设有 n 台超级洗衣机放在同一排上。开始的时候,每台洗衣机内可能有一定量的衣服,也可能是空的。
在每一步操作中,你可以选择任意 m (1 ≤ m ≤ n) 台洗衣机,与此同时将每台洗衣机的一件衣服送到相邻的一台洗衣机。
给定一个非负整数数组代表从左至右每台洗衣机中的衣物数量,请给出能让所有洗衣机中剩下的衣物的数量相等的最少的操作步数。如果不能使每台洗衣机中衣物的数量相等,则返回 -1。
示例 1:
输入: [1,0,5] 输出: 3 解释: 第一步: 1 0 <-- 5 => 1 1 4 第二步: 1 <-- 1 <-- 4 => 2 1 3 第三步: 2 1 <-- 3 => 2 2 2
示例 2:
输入: [0,3,0] 输出: 2 解释: 第一步: 0 <-- 3 0 => 1 2 0 第二步: 1 2 --> 0 => 1 1 1
示例 3:
输入: [0,2,0] 输出: -1 解释: 不可能让所有三个洗衣机同时剩下相同数量的衣物。
提示:
- n 的范围是 [1, 10000]。
- 在每台超级洗衣机中,衣物数量的范围是 [0, 1e5]。
分析:
如果洗衣机的衣服可以均分,那么最后只有两个情况:
1、当前位置之前洗衣机内所有衣物之和挪动到另外一边。
2、当前位置洗衣机内的衣物向两边挪动。
这两个情况都找到整个数组的最大值就是最后的结果。
代码:
1 class Solution { 2 public int findMinMoves(int[] machines) { 3 int sum=0; 4 for(int n:machines) 5 sum+=n; 6 int ave=sum/machines.length; 7 if(ave*machines.length<sum) 8 return -1; 9 sum=0; 10 int max=0; 11 for(int n=0;nn) { 12 sum+=machines[n]; 13 max=Math.max(max, Math.abs(sum-ave*(n+1))); 14 max=Math.max(max, machines[n]-ave); 15 } 16 return max; 17 } 18 }