快速求解霍夫曼编码树的带权路径长度

一、简介

霍夫曼编码被广泛的用在文本压缩编码中，但在求解霍夫曼树的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);
        }
    }
}