LeetCode517-超级洗衣机

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
解释：
不可能让所有三个洗衣机同时剩下相同数量的衣物。

提示：

• n == machines.length
• 1 <= n <= 104
• 0 <= machines[i] <= 105

贪心

1. 我们的最终目标就是让每个机器里的衣服数目一样，先排除不能整除的情况total % n != 0，直接返回-1；

2. 拿到最终平均值，avg = total / n；

3. $$定义\text{machines}[i{]}^{\prime }=\text{machines}[i]-\text{avg}$$

   将遍历过的机器看成A组，未遍历过的看成B组
   $$设\text{sum}[i]=\sum _{j=0}^i\text{machines}[j{]}^{\prime }$$
   sum[i]为正表示A组衣服多于平均，需要向B组转移；为负则需要B组给A组转移；

4. 然后取A、B之间需要移动衣服的次数：Math.abs(num) 和 某一台衣服过多的机器需要转移的衣服的更大值；

   // num = machines[i] - avg
   Math.max(Math.abs(sum), num)
   

解释：最开始看不懂的点有两个，一个是A组的（已经遍历过的）sum代表了什么，还有一个是sum为什么还需要和num比较？

• 对于第一点，我看到这个形象解释的评论：我们从左到右进行累加的sum就是把当前i节点左边的所有需要的操作全部转移到当前节点上，看成一个group A，sum < 0，需要后面的点转移进A；sum > 0,需要转出。

  思路: 有四个洗衣机，装的衣服数为[0, 0, 11, 5]，最终的状态会变为[4, 4, 4, 4]，那么我们将二者做差，得到[-4, -4, 7, 1]，这里负数表示当前洗衣机还需要的衣服数，正数表示当前洗衣机多余的衣服数。我们要做的是*要将这个差值数组每一项都变为0，对于第一个洗衣机来说，需要四件衣服可以从第二个洗衣机获得，那么就可以 把-4移给二号洗衣机，那么差值数组变为[0, -8, 7, 1]，此时二号洗衣机需要八件衣服，那么至少需要移动8次。然后二号洗衣机把这八件衣服从三号洗衣机处获得，那么差值数组变为[0, 0, -1, 1]，此时三号洗衣机还缺1件，就从四号洗衣机处获得，此时差值数组成功变为了[0, 0, 0, 0]，成功。那么移动的最大次数就是差值 数组中出现的绝对值最大的数字，8次*

• sum为什么还需要和num比较？num为什么不需要取绝对值？

  这一步是为了找max（machines元素）- avg，只有遍历过程中当前点的machines[i] - avg为正，当前点可以向左右节点进行衣服转移，这个步骤必须是一步一步进行的，所以
  num = machines[i] - avg就是要操作的步骤数，而machines[i] - avg为负，每一步操作中当前节点转移进来的衣服可以是两边也可能是一边，操作步骤数<=machines[i] - avg，所以不考虑。

class Solution {
    public int findMinMoves(int[] machines) {
        int ans = 0;
        int n = machines.length;
        int total = Arrays.stream(machines).sum();
        if (total % n != 0) return -1;
        int avg = total / n;
        int sum = 0;
        for (int num: machines) {
            num -= avg;
            sum += num;
            ans = Math.max(ans, Math.max(Math.abs(sum), num));
        }

        return ans;
    }
}