1. 程式人生 > >517. 超級洗衣機

517. 超級洗衣機

假設有 臺超級洗衣機放在同一排上。開始的時候,每臺洗衣機內可能有一定量的衣服,也可能是空的。

每一步操作中,你可以選擇任意 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]。

class Solution {
public:
    int findMinMoves(vector<int>& machines) {
        int len = machines.size();
        vector<int> sum(len + 1, 0);
        for (int i = 0; i < len; ++i)
            sum[i + 1] = sum[i] + machines[i];
 
        if (sum[len] % len) return -1;
 
        int avg = sum[len] / len;
        int res = 0;
        for (int i = 0; i < len; ++i)
        {
            int l = i * avg - sum[i];
            int r = (len - i - 1) * avg - (sum[len] - sum[i] - machines[i]);
 
            if (l > 0 && r > 0)
                res = std::max(res, std::abs(l) + std::abs(r));
            else
                res = std::max(res, std::max(std::abs(l), std::abs(r)));
        }
        return res;
    }
};