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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/super-washing-machines
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

1. 因为需要平均分配，所以总数 tot 一定能被洗衣机总数 n 整除，如果不能整除直接返回 -1；
2. ans 表示前 i 台洗衣机平均分配时需要的总操作数，sum 表示前 i 台洗衣机需要拿出或者拿进的衣服数量，num 表示第 i 台洗衣机需要拿出或者拿进衣服的数量。
3. ans 每次去三个的最大值。

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