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
解释:
不可能让所有三个洗衣机同时剩下相同数量的衣物。
提示:
- n 的范围是 [1, 10000]。
- 在每台超级洗衣机中,衣物数量的范围是 [0, 1e5]。
题解
贪心
我们先统计衣服数量的总和sum,显然当sum是洗衣机数量的整数倍时,才能使每台洗衣机的衣服数量相等,我们求出平均值
我们让每个洗衣机的衣服数量减去这个平均数,表明它缺少或多出几件衣服。由于每一步操作中一台洗衣机只能拿出一件衣服,因此如果洗衣机中衣服数x为正数,那么操作步数至少为x。对于前i台洗衣机,它们对应的数和若为y,那么它们需要和后n-i台洗衣机至少传递|y|次,操作步数至少为y。因此为了使当前状态所有洗衣机衣服数量相等,需要max(machines[i],currMax)次,其中currMax是对于0~i的|y|的最大值。详细过程见代码
代码
int findMinMoves(vector<int>& machines) {
int sum = 0,n = machines.size();
for(int i=0; i<n; i++)
sum += machines[i];
if(sum % n != 0) return -1;
int num = sum/n;
for(int i=0; i<n; i++)
machines[i] -= num;
int ans=0,curr=0,currMax=0;
for(int i=0; i<n; i++){
curr += machines[i];
currMax = max(currMax,abs(curr));
ans = max(ans,max(machines[i],currMax));
}
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/super-washing-machines
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。