数据结构与算法之美_37_贪心算法:如何用贪心算法实现霍夫曼树压缩编码?

痞子三分冷 提交于 2020-08-12 03:02:03

基础的数据结构和算法我们基本上已经学完了,接下来几节,我们学习更加基本的算法。它们分别是贪心算法、分治算法、回溯算法、动态规划。更加确切的说,它们应该是算法思想,并不是具体的算法,常用来指导我们设计具体的算法和编码等。

贪心、分治、回溯、动态规划这 4 个算法思想,原理解释起来很简单,但是要真正掌握且灵活应用,并不是件容易的事情。

贪心算法(greedy algorithm)有很多经典的应用,比如霍夫曼编码(Huffman Coding)、Prim 和 Kruskal 最小生成树算法、还有 Dijkstra 单源最短路径算法。今天学习下霍夫曼编码,看看它是如何利用贪心算法实现堆数据压缩编码,有效节省数据存储空间的。

如何理解“贪心算法”?

先看一个例子。假设我们有一个可以容纳 100kg 物品的背包,可以装各种物品。我们有以下 5 种豆子,每种豆子的总量和总价值都不相同。为了让背包中所装物品的总价最大,我们如何选择在背包中装哪些豆子呢?每种豆子又该装多少呢?

image

针对这个问题,我们先算法以下每个物品的单价,按照单价由高到低依次来装就好了。单价从高到低排列,依次是:黑豆、绿豆、红豆、青豆、黄豆,所以,我们可以往背包里装 20kg 黑豆、30kg 绿豆、50kg 红豆。

这个问题的解决思路显而易见,它本质上借助的就是贪心算法。结合这个例子,我们总结贪心算法解决问题的步骤。

第一步,当我们看到这类问题的时候,首先要联想到贪心算法:针对一组数据,我们定义了限制值和期望值,希望从中选出几个数据,在满足限制值的情况下,期望值最大。

类比刚刚的例子,限制值就是重量不能超过 100kg,期望值就是物品的总价值。这组数据就是 5 种豆子。我们从中选出一部分,满足重量不超过 100kg,并且总价值最大。

第二步,我们尝试看下这个问题是否可以用贪心算法解决:每次选择当前情况下,在对限制值相同贡献量的情况下,对期望值贡献量最大的数据。

类比刚才的例子,我们每次都从剩下的豆子里面,选择单价最高的,也就是重量相同的情况下,对价值贡献量最大的豆子。

第三步,我们举几个例子看下贪心算法产生的结果是否是最优的。大部分情况下,举几个例子验证一下就可以了。严格地证明贪心算法的正确性,是非常复杂的,需要涉及比较多的数学推理。

实际上,用贪心算法解决问题的思路,并不总能给出最优解。

我们举一个例子。在一个有权图中,我们从顶点 S 开始,找一条到顶点 T 的最短路径(路径中边的权值和最小)。贪心算法的解决思路是,每次都选择一条当前顶点相连的权最小的边,直到找到顶点 T。按照这种思路,我们求出的最短路径是 S->A->E->T,路径长度是 1+4+4=9。

image

但是,这种贪心算法的选择方式,最终求得的路径并不是最短路径,因为路径 S->B>D>T 才是最短路径,因为这条路径的长度是 2+2+2=6。为什么贪心算法在这个问题上不工作了呢?

在这个问题上,贪心算法不工作的主要原因是,前面的选择,会影响后面的选择。如果我们第一步从顶点 S 走到顶点 A,那接下来面对的顶点和边,跟第一步从顶点 S 走到顶点 B,是完全不同的。所以,即便我们第一步选择最优的走法(边最短),但有可能因为这一步选择,导致后面每一步的选择都很糟糕,最终也无缘全局最优解了。

贪心算法实战分析

掌握贪心算法的关键是多练习。

1. 分糖果

我们有 m 个糖果和 n 个孩子。现在要把糖果分给这些孩子吃,但是糖果少,孩子多(m < n),所以糖果只能分配给一部分孩子。

每个糖果的大小不等,这个 m 个糖果的大小分别是 s1,s2,s3,……,sm。除此之外,每个孩子对糖果的大小需求也是不一样的,只有糖果的大小大于等于孩子对糖果大小的需求的时候,孩子才能得到满足。假设这 n 个孩子对糖果的大小需求分别是 g1,g2,g3,……,gn。

我们的问题是,如果分配糖果,能尽可能满足最多数量的孩子?

我们可以把问题抽象成,从 n 个孩子中,抽取一部分孩子分配糖果,让满足的孩子的个数(期望值)是最大的。这个问题的限制值就是糖果个数m。

我们现在来看看如何用贪心算法来解决。对于一个孩子来说,如果小的糖果可以满足,我们就没必要用更大的糖果,这样更大的糖果就可以留给其他对糖果大小要求更大的孩子。另一方面,对糖果大小要求小的孩子更容易被满足,所以,我们可以从需求小的孩子开始分配糖果。因为满足一个需求大的孩子和满足一个需求小的孩子,对我们的期望值的贡献是一样的。

我们每次从剩下的孩子中,找出对糖果大小需求最小的,然后发给他剩下的糖果中能满足他的最小的糖果,这样得到的分配方案,也就是满足的孩子个数最多的方案。

2. 钱币找零

这个问题在我们日常生活中很普遍。假设我们有 1 元、2 元、5 元、10 元、20 元、50 元、100 元这些面额的纸币,它们的张数分别是

  • c1、c2、c5、c10、c20、c50、c100。我们现在要用这些钱来支付 K 元,最少需要用多少张纸币呢?

在生活中,我们肯定是先用面值最大的来支付,如果不够,就继续用更小一点面值的,以此类推,最后剩下的用 1 元来补齐。

我们分析下期望值、限制值、贡献值,期望值是纸币数量,限制值是 K 元。

在贡献相同期望值(纸币数目)的情况下,我们希望多贡献点金额,这样可以让纸币数更少,这就是一种贪心算法的解决思路。直觉告诉我们,这种处理方法就是最好的。实际上,更严谨地证明这种贪心算法的正确性,需要比较复杂的、有技巧的数学推导。

3. 区间覆盖

假设我们有 n 个区间,区间的起始端点和结束端点分别是 [l1, r1],[l2, r2],[l3, r3],……,[ln,rn]。我们从这 n 个区间中选出一部分区间,这部分区间满足两两不相交(端点相交的情况不算相交),最多能选出多少个区间呢?

image

这个问题的处理思路稍微不是那么好懂,不过,我们还是要弄懂它!因为这个思想在很多贪心算法问题上都有用到,比如任务调度、教师排课等等问题。

这个问题的解决思路是:我们假设这个 n 个区间中最左端点时 lmin,最右端点是 rmax。这个问题就相当于,我们选择几个不相交的区间,从左到右将 [lmin, rmax] 覆盖上。我们按照起始端点从小到大的顺序对这 n 个区间排序。

我们每次选择,左端点跟前面的已经覆盖的区间不重合的,右端点又尽量小的区间,这样可以让剩下的未覆盖区间尽可能的大,就可以放置更多的区间。这实际上就是一种贪心的选择方法。

image

解决开篇

如何用贪心算法实现霍夫曼编码?

霍夫曼编码是一种借助树形结构来生成的编码。

假设我们有一个包含 1000 个字符的文件,每个字符占 1 个 byte(8 bits),存储这 1000 个字符一共需要 8000 bits,那有没有更加节省空间的存储方式呢?

假设我们通过统计分析发现,这 1000 个字符中只包含 6 种不同字符,假设它们分别是 a、b、c、d、e、f。而 3 个二进制位(bit)就可以表示 8 个不同的字符,所以,为了尽量减少存储空间,每个字符我们用 3 个二进制位来表示。那存储这个 1000 个字符只需要 3000 bits 就可以了,比原来的存储方式节省了很多的空间。不过,还有没有更加节省空间的存储方式呢?

a(000)、b(001)、c(010)、d(011)、e(100)、f(101)

霍夫曼编码就要登场了!霍夫曼编码是一种十分有效的编码方式,广泛用于数据压缩中,其压缩率通常在 20% ~ 90% 之间。

霍夫曼不仅会考察文本中有多少个不同字符,还会考察每个字符出现的频率,根据频率的不同,选择不同长度的编码。霍夫曼编码试图用这种不等长的编码方式,来进一步增加压缩的效率。

如何给不同频率的字符选择不同长度的编码呢?根据贪心的思想,我们可以把出现频率比较多的字符,用稍微短一些的编码;出现频率比较少的字符,用稍微长一些的编码。

对于等长的编码来说,我们解压缩起来很简单。比如刚才那个例子中,我们用 3 个 bit 表示一个字符。在解压缩的时候,我们每次从文本中读取 3 位二进制码,然后翻译成对应的字符。但是,霍夫曼编码不是等长的,每次应该读取 1 位还是 2 位、3 位等等来解压缩呢?

这个问题就导致霍夫曼编码解压缩起来比较复杂,为了避免解压缩过程中的歧义,霍夫曼编码要求各个字符的编码之间,不会出现某个编码是另一个编码前缀的情况。

image

假设这 6 个字符出现的频率从高到低依次是 a、b、c、d、e、f。我们把它们编码成下面这个样子,任何一个字符的编码都不是另一个的前缀,在解压缩的时候,我们每次都会读取尽可能长的可解压的二进制串,所以在解压缩已的时候也不会歧义(进行读取时,记录一个当前读取位置,按照二进制编码是否为 1 为一个判定条件,判定读取的编码是一个可解压的编码)。经过这种编码压缩之后,这 1000 个字符只需要 2100 bits 就可以了。

image

霍夫曼编码的思想并不难理解,但是如何根据字符出现频率的不同,给不同的字符进行不同长度的编码呢?这里的处理有些技巧。

如下图所示中,我们把每个字符看做一个节点,并且辅带着把频率一起放到优先级队列中(构建成一个小顶堆,频率低的字符在堆顶),我们从队列中取出频率最小的两个节点 f、e,然后新建一个节点 x,把频率设置为两个节点的频率之和,并把这个新节点 x 作为节点 f、e 的父节点(构建成一棵二叉树)。最后再把 x 节点放入到优先级队列中。重复这个过程,直到队列没有数据。

image

现在,我们给每一条边画一个权值,指向左子节点的边我们统统标记为 0,指向右子节点的边,我们统统标记为 1,那从根节点到叶节点的路径就是叶节点对应字符的霍夫曼编码。

image

> 注意上图是一个特殊的例子,一般的霍夫曼树是下面的这样。

image

内容小结

今天我们学习贪心算法。实际上,贪心算法适用的场景比较有限。这种算法思想更多的是指导设计基础算法。比如最小生成树、单源最短路径算法,这些算法都用到了贪心算法。

贪心算法最难的一块是如何将要解决的问题抽象成为贪心算法模型,搞定这一步之后,贪心算法的编码一般都很简单。贪心算法解决问题的正确性虽然很多时候都看起来是显而易见的,但是要严谨地证明算法能够得到最优解,并不是容易的事情。所以,很多时候,我们只需要多举几个例子,看下贪心算法的解决方案是否真的能够最优解就可以了。

课后思考

  1. 在一个非负整数 a 中,我们希望从中移除 k 个数字,让剩下的数字值最小,如何选择移除哪 k 个数字呢?

答:从高位遍历整数字符,当高位的字符数字大于次高位的字符数时,将高位的树移除,并记录数字 k-1,如果高位字符小于等次高位字符数则不做处理,继续往后遍历,直到移除 k 个数字。

如果整个字符数遍历完一遍,但是移除的数字 m 个(m < k),则移除字符数组的后 k - m 位。

  1. 假设有 n 个人等待被服务,但是服务窗口只有一个,每个人需要被服务的时间长度是不同的,如何安排被服务的先后顺序,才能让这个 n 个人等待的时间最短呢?

答:根据需要被服务的时间长度从小到大的顺序服务这 n 个人。

代码

Github

参考资料

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!