【面试高频题】难度 4/5，经典贪心运用及其证明

原创

宫水三叶 2023-03-26 04:45:34 ©著作权

文章标签 后端 Java 算法 Math github 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者宫水三叶的原创作品，请联系作者获取转载授权，否则将追究法律责任

题目描述

这是 LeetCode 上的 517. 超级洗衣机 ，难度为困难。

Tag : 「贪心」

假设有 n 台超级洗衣机放在同一排上。开始的时候，每台洗衣机内可能有一定量的衣服，也可能是空的。

在每一步操作中，你可以选择任意 m (1 <= m <= n) 台洗衣机，与此同时将每台洗衣机的一件衣服送到相邻的一台洗衣机。

给定一个整数数组 machines 代表从左至右每台洗衣机中的衣物数量，请给出能让所有洗衣机中剩下的衣物的数量相等的最少的操作步数。如果不能使每台洗衣机中衣物的数量相等，则返回 -1 。

示例 1：

输入：machines = [1,0,5]

输出：3

解释：
第一步:    1     0 <-- 5    =>    1     1     4
第二步:    1 <-- 1 <-- 4    =>    2     1     3
第三步:    2     1 <-- 3    =>    2     2     2

示例 2：

输入：machines = [0,3,0]

输出：2

解释：
第一步:    0 <-- 3     0    =>    1     2     0
第二步:    1     2 --> 0    =>    1     1     1

示例 3：

输入：machines = [0,2,0]

输出：-1

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

提示：

$【面试高频题】难度 4/5，经典贪心运用及其证明_算法$
$【面试高频题】难度 4/5，经典贪心运用及其证明_Java_02$
$【面试高频题】难度 4/5，经典贪心运用及其证明_github_03$

基本分析

由于最终是要让所有洗衣机衣服相等，因此无解的情况很好分析，如果衣服数量 $【面试高频题】难度 4/5，经典贪心运用及其证明_Java_04$ 不能整除洗衣机数量 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_05$ 的话，则返回 $【面试高频题】难度 4/5，经典贪心运用及其证明_Math_06$ ，否则必然有解（最坏情况下，每次只移动一件衣服，也可以使得衣服均分），要求最小移动次数。

由于每次操作都可以选任意台机器进行，不难猜想到最小移动次数为 所有机器的「最小运输衣服数量」中的最大值。

计算某台洗衣机的「最小运输衣服数量」为经过当前机器的衣服数量（每次只能运输一件衣服），其值等于「起始左边衣服总量与最终左边衣服总量的差值」+「起始右边衣服总量与最终右边衣服总量的差值」，这里的差值都需要与 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ 取 $【面试高频题】难度 4/5，经典贪心运用及其证明_Java_08$

我们猜想取所有机器中的「最小操作次数」的最大值即是答案。

但这显然是理论的最小操作次数，我们来证明最终答案等于该值。

假设理论最下操作次数为 $【面试高频题】难度 4/5，经典贪心运用及其证明_Java_09$ ，真实答案为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ ，那么天然有 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_11$ ，我们需要通过证明 $【面试高频题】难度 4/5，经典贪心运用及其证明_github_12$ 恒成立，来得证 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_13$ 。

可以通过「反证法」来证明 $【面试高频题】难度 4/5，经典贪心运用及其证明_github_12$ 恒成立，假设 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_15$ ，即在某个特定序列中，实际最小操作次数 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ 大于 $【面试高频题】难度 4/5，经典贪心运用及其证明_Java_09$ ，假定我们是在位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$

那么我们需要思考：在没有无效传输的前提，什么情况下需要在 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 位置传输大于 $【面试高频题】难度 4/5，经典贪心运用及其证明_Java_09$

注：无效的意思是，衣服从位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$

（注 1）当且仅当位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 本身衣服为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$

也就是说首次传输，并没有实现「从 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 左边往右边传输衣服」或者「从 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 右边往左边传输衣服」的目的，而是需要先往位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$

那么是否可能由起始衣服为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ 的位置来取得 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ 呢？我们通过「反证法」来证明「 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ 不可能由衣服为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$

由于位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 的起始数量为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ ，那么位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$

如果是两边都多于最终数量，说明最终是两边衣服流向位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ ，而且我们得到的 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ 是两边的缺少总和，这种情况下得到的 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ 为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ ，但是整体衣服本身不相等，必然要消耗步数，必然不为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ ，因此该情况不存在。

既然位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 至少有一侧的起始数量小于最终数量的（缺少衣服的），那么自然我们可以将位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 归到那一边，使得那一侧缺少衣服的数量更多，从而使答案 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ 更大。这与 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$

得证，取得 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ 的位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 起始衣服必然不为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ 。

如果位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 起始衣服必然不为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ ，那么（注 1）的条件不成立，则 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_15$ 恒不成立，得证 $【面试高频题】难度 4/5，经典贪心运用及其证明_github_12$

至此，我们通过三次「反证法」来证明了结论成立。首先通过「反证法」证明取得 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_10$ 的位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 衣服不可能为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ ；然后根据该位置起始衣服不为 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ 的前提条件，来证明 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_15$ 恒不成立，得证 $【面试高频题】难度 4/5，经典贪心运用及其证明_github_12$ 恒成立，最终结合 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_11$ 来得证 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_13$ 。

贪心

实现上，首先我们可以求得衣服总和 $【面试高频题】难度 4/5，经典贪心运用及其证明_Java_04$ 以及洗衣机数量 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_05$ ，从而判断无解情况（sum % n != 0），或者计算最终每台洗衣机的衣服数量 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_60$ 。

然后使用两个变量 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_61$ 和 $【面试高频题】难度 4/5，经典贪心运用及其证明_github_62$

对于某个位置 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 而言，达到最终平衡需要从 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 右边往左边运送的衣服数量为 $【面试高频题】难度 4/5，经典贪心运用及其证明_Math_65$ ，即左边的当前的衣服数量与最终状态的衣服数量的差值，与 $【面试高频题】难度 4/5，经典贪心运用及其证明_算法_07$ 取 $【面试高频题】难度 4/5，经典贪心运用及其证明_Java_08$ 含义代表为如果当前左边衣服多于最终衣服数量时，此时不需要消耗从右到左的移动次数（只需要消耗从 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 左边到 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 右边的移动次数）；右边分析同理，我们可以得到达到最终平衡需要从 $【面试高频题】难度 4/5，经典贪心运用及其证明_后端_18$ 左边到右运送的衣服数量为 $【面试高频题】难度 4/5，经典贪心运用及其证明_Math_71$ 。

在所有位置的 $【面试高频题】难度 4/5，经典贪心运用及其证明_github_72$

代码：

class Solution {
    public int findMinMoves(int[] ms) {
        int n = ms.length;
        int sum = 0;
        for (int i : ms) sum += i;
        if (sum % n != 0) return -1;
        int t = sum / n;
        int ls = 0, rs = sum;
        int ans = 0;
        for (int i = 0; i < n; i++) {
            rs -= ms[i];
            int a = Math.max(t * i - ls, 0);
            int b = Math.max((n - i - 1) * t - rs, 0);
            ans = Math.max(ans, a + b);
            ls += ms[i];
        }
        return ans;
    }
}