LeetCode 517 超级洗衣机 解法

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/super-washing-machines
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
题目描述：
假设有 n 台超级洗衣机放在同一排上。开始的时候，每台洗衣机内可能有一定量的衣服，也可能是空的。
在每一步操作中，你可以选择任意 m （1 ≤ m ≤ n） 台洗衣机，与此同时将每台洗衣机的一件衣服送到相邻的一台洗衣机。
给定一个非负整数数组代表从左至右每台洗衣机中的衣物数量，请给出能让所有洗衣机中剩下的衣物的数量相等的最少的操作步数。如果不能使每台洗衣机中衣物的数量相等，则返回 -1。

示例 1：
输入: [1,0,5]
输出: 3
解释:
第一步: 1 0 <-- 5 => 1 1 4
第二步: 1 <-- 1 <-- 4 => 2 1 3
第三步: 2 1 <-- 3 => 2 2 2

示例 2：
输入: [0,3,0]
输出: 2
解释:
第一步: 0 <-- 3 0 => 1 2 0
第二步: 1 2 --> 0 => 1 1 1

示例 3:
输入: [0,2,0]
输出: -1
解释:
不可能让所有三个洗衣机同时剩下相同数量的衣物。
提示：
n 的范围是 [1, 10000]。
在每台超级洗衣机中，衣物数量的范围是 [0, 1e5]。

解题思路：
首先，先分析可以成立的情况，如果 衣服数量%机器总数 != 0，那么无法做到每一台机器都具有相同的衣服，所以我们只需要分析能做到每台机器都具有平均数量衣服的情况。
接下来先求出这个平均值，设为k。
以机器中的某一个点i为例子进行分析，设i左边有a台机器，右边有b台机器，求出k * a - sum左和k * a - sum右，如果小于0，说明这个区域多了衣服，若大于0，说明这个区域少了衣服。

设两个差值分别为a和b，基于这2个值可以进行讨论：
1）如果a > 0，b > 0 ，则两边都少了衣服，说明i点多出的衣服需要向左扔，这个时候需要进行a+b轮衣服去扔，因为每次只能扔一件衣服；
2）如果a < 0，b < 0，则两边都多了衣服，说明i点少了衣服，需要左边和右边去扔衣服，这个时候需要进行max(|a|, |b|)轮，因为是两边同时扔衣服，所以瓶颈在于较大的值；
3）如果a < 0，b > 0 或 a > 0，b < 0，说明一边缺衣服，一边多衣服，这个时候也需要进行max(|a|, |b|)轮，因为有一边需要去扔衣服，一边需要去接衣服，两个可以并行进行，瓶颈还是取决于最大值。
基于以上的策略，我们便可以将小的推论推广，从头到位去算每台机器左边和右边的瓶颈，那么其中最大的瓶颈就是我们要求的最小操作步数。因为一旦你有一个机器的操作步数大于其它的机器，说明最终的最小步数还是会受到这个的限制。
另外，每次遍历的时候都需要去求sum左和sum右，我们可以先通过预处理数组的方式把sum求出来，这样每次求和求可以直接通过下标访问即可。
基于以上的分析，可以写出下列代码（C++）：

class Solution {
public:
    int findMinMoves(vector<int>& machines) {
        if (machines.size() < 1) {
            return -1;
        }
        int size = machines.size();
        int sum = 0;
        for (int i = 0; i < size; i++) {
            sum += machines[i]; //生成预处理数组
        }
        if (sum % size != 0) {
            return -1;
        }
        int avg = sum / size;
        int leftSum = 0;
        int res = 0;
        for (int i = 0; i < size; i++) {
            int L = i * avg - leftSum;//i左边的需要的数目-累加和
            int R = (size - i - 1) * avg - (sum - leftSum - machines[i]);//i右边的需要数目-累加和
            if (L > 0 && R > 0) { //如果L > 0，R > 0 ，则两边都少了衣服，瓶颈在于两者之和
                res = std::max(L + R, res);
            } else { //其它情况的瓶颈都在于两者的绝对值的最大值
                res = std::max(std::max(std::abs(L), std::abs(R)), res);
            }
            leftSum += machines[i];
        }
        return res;
    }
};