java 哈夫曼树_Java--哈夫曼树创建和遍历

###哈夫曼树的创建和遍历

import java.util.*;

public class huffman

{

public static void main(String[] args)

{

int []arr={13,7,8,3,29,6,1};

Node root=huff(arr);

//前序遍历打印赫夫曼树

if(root!=null){

System.out.println("赫夫曼树前序遍历为:");

root.preOrder();}

else{

System.out.println("赫夫曼树为空");

}

}

//创建赫夫曼树

public static Node huff(int []arr){

//创建Node类型的集合，便于存取节点

Listnodes=new ArrayList();

for(int a:arr){

nodes.add(new Node(a));//注意：此处是把Node类型节点放入，所以构造新的对象

}

while(nodes.size()>1){

//排序，小到大

Collections.sort(nodes);//对集合内部元素排序

//System.out.println(nodes);

//(1)取两个节点最小的构建子树

Node left=nodes.get(0);

Node right=nodes.get(1);

//(2)构造父节点

Node parent=new Node(left.value+right.value);

parent.left=left;

parent.right=right;

//(3)从集合中删除已经取出的节点

nodes.remove(left);

nodes.remove(right);

//(4)将父节点加入到集合中

nodes.add(parent);//默认在集合最后

}

return nodes.get(0);

}

}

class Node implements Comparable

{

int value;

Node left;

Node right;

public Node(int value){

this.value=value;

}

public String toString(){

return "value="+value;

}

public int compareTo(Node o){

return this.value-o.value;//实现从小到大排序

}

//前序遍历赫夫曼树

public void preOrder(){

System.out.println(this);

if(this.left!=null){

this.left.preOrder();

}

if(this.right!=null){

this.right.preOrder();

}

}

}