517. 超级洗衣机[贪心]

• 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

class Solution {
    public int findMinMoves(int[] machines) {
        int n=machines.length;
        int sum=0;
        for(int num:machines){
            sum+=num;
        }
        if(sum%n!=0)return -1;

        int avg=sum/n;
    
        int ans=0;
        int res=0;
        for(int x:machines){
            int num=x-avg;
            res+=num;
            ans=Math.max(Math.abs(res),Math.max(ans,num));
        }
        return ans;
    }
}

贪心：
如果没有相邻的限制，那么答案是max(machines[i]-avg)
因为洗衣机只能在相邻之间移动，但是可以直接把需要减少的给到需要增加的里，但是中间的洗衣机少了一次调整的机会，所以出现了sum变量