Leetcode 517. 超级洗衣机 C++

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]。

题解

贪心
我们先统计衣服数量的总和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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。