Java实现 LeetCode 517 超级洗衣机

港控/mmm° 2023-07-17 15:51 20阅读 0赞

## 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

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

提示：

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

PS：  
我移动的最小的次数，无非来源于两种  
我当前值和平均值的差  
我前面差的和

class Solution { 
        public int findMinMoves(int[] machines) { 
            int sum = 0;
            for(int num : machines)
              sum += num;
            if(sum % machines.length != 0)
                return -1;
    
            int target = sum / machines.length;
            int res = 0, balance = 0;
            for(int i = 0 ; i < machines.length; i ++){ 
                balance += machines[i] - target;
                res = Math.max(res, Math.max(machines[i] - target, Math.abs(balance)));
            }
            return res;
        }
    }