Huffman树的简单实现

1、定义
给定n个权值作为n的叶子结点,构造一棵二叉树,若带权路径长度达到最小,称这样的二叉树为最优二叉树,也称为哈夫曼树(Huffman Tree)。哈夫曼树是带权路径长度最短的树,权值较大的结点离根较近。
2、实现思想
树节点放入优先队列,每次取出两个最小的构造新节点,并把新节点加入队列,直到队列只剩下一个元素,此元素即为构造的树的树根。(学会使用抽象数据结构来实现更加高级的功能)。
Java代码如下:


import java.util.PriorityQueue;
import java.util.Queue;

/**
 * Created by lemon on 2016/4/13.
 */
/*定义树节点*/
class Node implements Comparable<Node> {
    public int val;
    public Node left;
    public Node right;
    public Node(int val){
        this.val=val;
        left=null;
        right=null;
    }
    @Override
    public int compareTo(Node o) {
        return this.val-o.val;
    }
}
public class HuffTree {
/*构造huffman树*/
    public static Node HuffTree(int [] array){
        Queue<Node> queue=new PriorityQueue<Node>();
        for(int i : array)
            queue.add(new Node(i));
        while(queue.size()>1){
            Node node1=queue.poll();
            Node node2=queue.poll();
            Node node3=new Node(node1.val+node2.val);
            node3.left=node1;
            node3.right=node2;
            queue.add(node3);
        }
        return queue.poll();
    }
    //先序遍历树
    public static void preOrder(Node root){
        if(root!=null){
            System.out.print(root.val+" ");
            preOrder(root.left);
            preOrder(root.right);
        }
    }
    public static void main(String args[]){
        int array[]={1,2,3,4,5,6,7,8};
        Node root=HuffTree(array);
        preOrder(root);
        System.exit(0);
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章