关于后端:面试高频题难度-45经典贪心运用及其证明

35次阅读

共计 2803 个字符,预计需要花费 8 分钟才能阅读完成。

题目形容

这是 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

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

提醒:

  • $n = machines.length$
  • $1 <= n <= 10^4$
  • $0 <= machines[i] <= 10^5$

根本剖析

因为最终是要让所有洗衣机衣服相等,因而无解的状况很好剖析,如果衣服数量 $sum$ 不能整除洗衣机数量 $n$ 的话,则返回 $-1$,否则必然有解(最坏状况下,每次只挪动一件衣服,也能够使得衣服均分),要求最小挪动次数。

因为每次操作都能够选任意台机器进行,不难猜想到最小挪动次数为 所有机器的「最小运输衣服数量」中的最大值

计算某台洗衣机的「最小运输衣服数量」为通过以后机器的衣服数量(每次只能运输一件衣服),其值等于「起始右边衣服总量 与 最终右边衣服总量 的差值」+「起始左边衣服总量 与 最终左边衣服总量 的差值」,这里的差值都须要与 $0$ 取 $\max$ 代指短少衣服的数量(因为如果是多余数量的话,能够通过同时传输来满足减少短少的一边,缩小多余的一边)。

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

但这显然是实践的最小操作次数,咱们来证实最终答案等于该值。

假如实践最下操作次数为 $cnt$,实在答案为 $ans$,那么人造有 $ans \geq cnt$,咱们须要通过证实 $ans \leq cnt$ 恒成立,来得证 $ans = cnt$。

能够通过「反证法」来证实 $ans \leq cnt$ 恒成立,假如 $ans > cnt$,即在某个特定序列中,理论最小操作次数 $ans$ 大于 $cnt$,假设咱们是在地位 $x$ 中获得这个理论最小操作次数。

那么咱们须要思考:在没有有效传输的前提,什么状况下须要在 $x$ 地位传输大于 $cnt$ 件衣服来达到最终均衡。

注:有效的意思是,衣服从地位 $x$ 的一边传到另外一边,随后又传输回来。

(注 1)当且仅当地位 $x$ 自身衣服为 $0$ 时,会产生该种状况。

也就是说首次传输,并没有实现「从 $x$ 右边往右边传输衣服」或者「从 $x$ 左边往左边传输衣服」的目标,而是须要先往地位 $x$ 填送衣服。

那么是否可能由起始衣服为 $0$ 的地位来获得 $ans$ 呢?咱们通过「反证法」来证实「$ans$ 不可能由衣服为 $0$ 的起始地位得出」。

因为地位 $x$ 的起始数量为 $0$,那么地位 $x$ 必然至多有一侧的起始数量小于最终数量的(短少衣服的),能够持续利用「反证法」来证实:

  • 如果是两边都多于最终数量,阐明最终是两边衣服流向地位 $x$,而且咱们失去的 $ans$ 是两边的短少总和,这种状况下失去的 $ans$ 为 $0$,然而整体衣服自身不相等,必然要耗费步数,必然不为 $0$,因而该状况不存在。

既然地位 $x$ 至多有一侧的起始数量小于最终数量的(短少衣服的),那么天然咱们能够将地位 $x$ 归到那一边,使得那一侧短少衣服的数量更多,从而使答案 $ans$ 更大。这与 $ans$ 为所有地位中的「最小操作次数」最大的地位矛盾。

得证,获得 $ans$ 的地位 $x$ 起始衣服必然不为 $0$。

如果地位 $x$ 起始衣服必然不为 $0$,那么(注 1)的条件不成立,则 $ans > cnt$ 恒不成立,得证 $ans \leq cnt$ 恒成立。

至此,咱们通过三次「反证法」来证实了论断成立。首先通过「反证法」证实获得 $ans$ 的地位 $x$ 衣服不可能为 $0$;而后依据该地位起始衣服不为 $0$ 的前提条件,来证实 $ans > cnt$ 恒不成立,得证 $ans \leq cnt$ 恒成立,最终联合 $ans \geq cnt$ 来得证 $ans = cnt$。

贪婪

实现上,首先咱们能够求得衣服总和 $sum$ 以及洗衣机数量 $n$,从而判断无解状况(sum % n != 0),或者计算最终每台洗衣机的衣服数量 $t = sum / n$。

而后应用两个变量 $ls$ 和 $rs$ 别离示意以后地位「右边的衣服总数」和「左边的衣服总数」,并在从左往右的遍历过程中实时保护。

对于某个地位 $x$ 而言,达到最终均衡须要从 $x$ 左边往左边运送的衣服数量为 $a = \max(i * t – ls, 0)$,即右边的以后的衣服数量与最终状态的衣服数量的差值,与 $0$ 取 $\max$ 含意代表为如果以后右边衣服多于最终衣服数量时,此时不须要耗费从右到左的挪动次数(只须要耗费从 $x$ 右边到 $x$ 左边的挪动次数);左边剖析同理,咱们能够失去达到最终均衡须要从 $x$ 右边到右运送的衣服数量为 $b = \max((n – i – 1) * t – rs, 0)$。

在所有地位的 $a + b$ 之间取最大值即是答案。

代码:

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;
    }
}
  • 工夫复杂度:$O(n)$
  • 空间复杂度:$O(1)$

最初

这是咱们「刷穿 LeetCode」系列文章的第 No.517 篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,局部是有锁题,咱们将先把所有不带锁的题目刷完。

在这个系列文章外面,除了解说解题思路以外,还会尽可能给出最为简洁的代码。如果波及通解还会相应的代码模板。

为了不便各位同学可能电脑上进行调试和提交代码,我建设了相干的仓库:https://github.com/SharingSou…。

在仓库地址里,你能够看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其余优选题解。

更多更全更热门的「口试 / 面试」相干材料可拜访排版精美的 合集新基地 🎉🎉

本文由 mdnice 多平台公布

正文完
 0