LeetCode——517. 超级洗衣机

题目描述：

假设有 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
解释:

• 不可能让所有三个洗衣机同时剩下相同数量的衣物。

提示：

1. n 的范围是 [1, 10000]。
2. 在每台超级洗衣机中，衣物数量的范围是 [0, 1e5]。

题解

代码如下：

class Solution {
    public int findMinMoves(int[] machines) {
        int n = machines.length;
        if (n == 0) {
            return 0;
        }
        int sum = 0;
        for (int machine : machines) {
            sum += machine;
        }
        if (sum % n != 0) {
            return -1;
        }
        int avg = sum / n;
        int leftSum = 0;
        int ans = 0;
        for (int i = 0; i < n; i++) {
            int leftRest = leftSum - avg * i;
            int rightRest = (sum - leftSum - machines[i]) - (n - i - 1) * avg;
            if (leftRest < 0 && rightRest < 0) {
                ans = Math.max(ans, Math.abs(leftRest) + Math.abs(rightRest));
            } else {
                ans = Math.max(ans, Math.max(Math.abs(leftRest), Math.abs(rightRest)));
            }
            leftSum += machines[i];
        }
        return ans;
    }
}

执行结果：