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);
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章