重温数据结构:Huffman树的建立
原创
©著作权归作者所有:来自51CTO博客作者ali莫川的原创作品,请联系作者获取转载授权,否则将追究法律责任
1.Huffman树
霍夫曼树又称最优二叉树,是一种带权路径长度最短的二叉树。所谓树的带权路径长度,就是树中所有的叶结点的权值乘上其到根结点的路径长度(若根结点为0层,叶结点到根结点的路径长度为叶结点的层数)。树的路径长度是从树根到每一结点的路径长度之和,记为WPL=(W1*L1+W2*L2+W3*L3+...+Wn*Ln),N个权值Wi(i=1,2,...n)构成一棵有N个叶结点的二叉树,相应的叶结点的路径长度为Li(i=1,2,...n)。可以证明霍夫曼树的WPL是最小的。由于《信息论》中我们学过Huffman编码,因此,对其基本概念是比较熟悉的。以下是Huffman树的建立过程,并用层次遍历检测建立的树是否正确。
package edu.njupt.zhb;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;
import org.w3c.dom.ls.LSException;
/*
*@author: ZhengHaibo
*2014-3-15 Nanjing,njupt,China
*/
public class MyHuffmanTree {
public class TreeNode{
int weight;
TreeNode left;
TreeNode right;
public TreeNode() {
// TODO Auto-generated constructor stub
}
}
/**
* 构建Huffman树
* @param nodes
* @return
*/
public TreeNode buildHuffmanTree(TreeNode []nodes){
List<TreeNode> list=new LinkedList<TreeNode>();
for(TreeNode treeNode:nodes){
insertIntoList(list, treeNode);//将节点按照权值进行排序
}
while(list.size()>1){//从有序链表中依次取值
TreeNode min1=list.remove(list.size()-1);//取出最小值
TreeNode min2=list.remove(list.size()-1);//取出当前最小值
TreeNode newTreeNode=new TreeNode();//生成新节点
newTreeNode.weight=min1.weight+min2.weight;
newTreeNode.left=min1;
newTreeNode.right=min2;
insertIntoList(list, newTreeNode);//将新节点插回有序链表中
}
return list.get(0);
}
/**
* 插入有序表
* 根据权值由大到小进行插入排序
*/
public void insertIntoList(List<TreeNode> list,TreeNode node){
if(list.size()==0){
list.add(node);
return;
}
TreeNode temp;
for(int i=0;i<list.size();i++){
if(node.weight>list.get(i).weight){
temp=list.get(i);
list.add(i,node);
return;
}
}
list.add(node);
}
/**
* 层次遍历打印
* @param root
*/
public void levelOrder(TreeNode root){
if(root==null){
return;
}
Queue queue=new LinkedBlockingQueue();
queue.add(root);
while(!queue.isEmpty()){
TreeNode node=(TreeNode)queue.remove();
System.out.println(node.weight);
if(node.left!=null){
queue.add(node.left);
}
if (node.right!=null) {
queue.add(node.right);
}
}
}
/**
* 测试
*/
public void test(){
int []weights={7,4,2,9,15,5};
TreeNode []nodes=new TreeNode[weights.length];
for(int i=0;i<weights.length;i++){
TreeNode node=new TreeNode();
node.weight=weights[i];
nodes[i]=node;
}
TreeNode root=buildHuffmanTree(nodes);
levelOrder(root);
}
public static void main(String[] args) {
new MyHuffmanTree().test();
}
}
以上涉及的huffman树,结构如下: