517. 超级洗衣机
假设有 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
解释:
不可能让所有三个洗衣机同时剩下相同数量的衣物。
解题思路
- 先计算出衣服的总数,判断衣服能否被平均分到每一台洗衣机里面
- 对于每一台洗衣机,我们可以计算出其与平均后衣服数量的差值,如果为负数,意味着需要抽调其他洗衣机的衣服过来,如果为正数,则是需要把当前衣服移除出去。我们将前面若干洗衣机分为一组,后面分为一组,维护一个变量sum,代表前一组洗衣机需要移动多少件衣服进入后一组才可以使得衣服的数量均等。因为每次只能移动一件衣服,因此如果要把前一组的移动到后一组来补充衣服的话,最少需要sum次移动,特殊情况是,当其他一个洗衣机衣服太多,不光要分去后一组,还要平摊到前一组的前部分,至少需要machines[i]-tar次移动
代码
class Solution {public int findMinMoves(int[] machines) {int n=machines.length,sum=0;for(int m:machines)sum+=m;if(sum%n!=0) return -1;int tar=sum/n,pre=0,res=0;for(int i=0;i<n;i++){machines[i]-=tar;pre+=machines[i];res=Math.max(res,Math.max(Math.abs(pre),machines[i]));} return res;}
}