假设有 n
台超级洗衣机放在同一排上。开始的时候,每台洗衣机内可能有一定量的衣服,也可能是空的。
在每一步操作中,你可以选择任意 m
(1 <= m <= n
) 台洗衣机,与此同时将每台洗衣机的一件衣服送到相邻的一台洗衣机。
给定一个整数数组 machines
代表从左至右每台洗衣机中的衣物数量,请给出能让所有洗衣机中剩下的衣物的数量相等的 最少的操作步数 。如果不能使每台洗衣机中衣物的数量相等,则返回 -1
。
示例 1:
输入:machines = [1,0,5] 输出:3 解释: 第一步: 1 0 <-- 5 => 1 1 4 第二步: 1 <-- 1 <-- 4 => 2 1 3 第三步: 2 1 <-- 3 => 2 2 2
示例 2:
输入:machines = [0,3,0] 输出:2 解释: 第一步: 0 <-- 3 0 => 1 2 0 第二步: 1 2 --> 0 => 1 1 1
示例 3:
输入:machines = [0,2,0] 输出:-1 解释: 不可能让所有三个洗衣机同时剩下相同数量的衣物。
提示:
n == machines.length
1 <= n <= 104
0 <= machines[i] <= 105
有点绕。
大意是,最大次数为要往某个区域内移入的最大次数,或者当前位置的洗衣机移出的最大次数,这两个值中较大的一个。
class Solution {
public int findMinMoves(int[] machines) {
int sum = 0;
int[] preSum = new int[machines.length+1];
for (int i = 0; i < machines.length; i++) {
preSum[i+1] = preSum[i] + machines[i];
}
sum = preSum[machines.length];
if (sum % machines.length != 0){
return -1;
}
int avg = sum/machines.length;
int ans = 0;
for (int i = 0; i < machines.length; i++) {
ans = Math.max(ans,Math.abs(preSum[i] - avg * i));
ans = Math.max(machines[i]-avg,ans);
}
return ans;
}
}