leetcode 517. 超级洗衣机

Posted 2021-12-02 chaha123

题目：

假设有 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]。

 

分析：

如果洗衣机的衣服可以均分，那么最后只有两个情况：

1、当前位置之前洗衣机内所有衣物之和挪动到另外一边。

2、当前位置洗衣机内的衣物向两边挪动。

这两个情况都找到整个数组的最大值就是最后的结果。

代码：

 1 class Solution {
 2     public int findMinMoves(int[] machines) {
 3         int sum=0;
 4         for(int n:machines)
 5             sum+=n;
 6         int ave=sum/machines.length;
 7         if(ave*machines.length<sum)
 8             return -1;
 9         sum=0;
10         int max=0;
11         for(int n=0;n<machines.length;++n) {
12             sum+=machines[n];
13             max=Math.max(max, Math.abs(sum-ave*(n+1)));
14             max=Math.max(max, machines[n]-ave);
15         }
16         return max;
17     }
18 }