LeetCode 517. 超级洗衣机

LeetCode 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 <= 10e4
0 <= machines[i] <= 10e5

思路

这题想了蛮久想不出来好办法，可以想到如果缺少衣服的洗衣机是要从旁边的多衣服的洗衣机拿来，一直想状态转移然后转移不出来，看题解看的也是亿头雾水。打开评论区看到一份不错的思路，可以说是醍醐灌顶，受教了：

解题思路: 有四个洗衣机，装的衣服数为[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次
思路来源：LeetCode用户 tracywwp

然后我们来看题解代码

class Solution {
public:
    int findMinMoves(vector<int> &machines) {
        int tot = accumulate(machines.begin(), machines.end(), 0);
        int n = machines.size();
        if (tot % n) {
            return -1;
        }
        int avg = tot / n;
        int ans = 0, sum = 0;
        for (int num: machines) {
            num -= avg;
            sum += num;
            ans = max(ans, max(abs(sum), num));
        }
        return ans;
    }
};

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/super-washing-machines/solution/chao-ji-xi-yi-ji-by-leetcode-solution-yhej/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

首先是求和取模判断能不能平方，很好理解。
然后是做差判断该洗衣机是缺了衣服还是多了衣服。

最核心的一步：ans = max(ans, max(abs(sum), num));
abs(num)：表示如果该位置缺少了衣服要从旁边拿衣服的次数
num：表示如果该位置多了衣服要给旁边衣服，且一次只能给一件，给完需要num次
从这三个数里不断取最大值就是答案

这个方法很巧妙，因为不用管其它洗衣机的状态，只需要看自己是多了还是少了衣服，判断这个洗衣机需要操作的次数，妙哉。

本日记仅供学习交流，侵删