一、简介
霍夫曼编码被广泛的用在文本压缩编码中,但在求解霍夫曼树的WPL长度时其实不必真的构造出一棵霍夫曼树再去实际求解,在实际oj和测试时时间也不允许,求解WPL可以基于如下定义快速求解:
WPL = 所有叶子节点的带权路径长度之和 = 霍夫曼树中所有非叶节点的权值之和
证明如图:
图片转载自https://blog.csdn.net/hchild1/article/details/50933972
如果是用第一个等式去做的化需要先构造出一棵霍夫曼树,再去求解所有叶子节点的带权路径,很麻烦!但是第二个等式就很容易了。
二、思路
将所有权重存入一个一个小顶堆中(可以直接使用Java中定义好的优先级队列),然后从队列中每次取出两个元素,将其和累加到初始化为0的WPL上,并再存入队列中,当队列中只有一个元素时退出循环!
三、Java实现
以POJ1521为例!
import java.util.*;
public class Main2 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
String str;
while (!(str = sc.next()).equals("END")) {
int n = str.length();
int asciiLen = n * 8;
Map<Character, Integer> memo = new HashMap<Character, Integer>();
for (char ch : str.toCharArray()) {
if (memo.containsKey(ch)) {
memo.put(ch, memo.get(ch) + 1);
} else {
memo.put(ch, 1);
}
}
Queue<Integer> pqueue = new PriorityQueue<Integer>();
pqueue.addAll(memo.values());
int prefixLen = 0;
if (pqueue.size() == 1) {
//只有一个节点,WPL = 1 * 该节点权重
prefixLen = pqueue.poll();
} else {
while (pqueue.size() > 1) {
int sum = pqueue.poll() + pqueue.poll();
prefixLen += sum;
pqueue.add(sum);
}
}
System.out.printf("%d %d %.1f\n", asciiLen, prefixLen, 1.0 * asciiLen / prefixLen);
}
}
}