实现步骤
- 首先将树节点进行排序,放入一个容器中
- 取出最小的两个树节点,制造出一个父节点,组成一个二叉树
- 在容器中删除最小的两个结点,将生成的父节点加入容器中
- 对容器进行重新排序,重复1-2-3操作,直到容器中只留下root结点(最终父节点或wpl的最终值)
定义二叉树结点的属性和方法
class Node implements Comparable<Node> {
int value;
char c;
Node left;
Node right;
public void preOrder() {
System.out.println(this);
if(this.left != null) {
this.left.preOrder();
}
if(this.right != null) {
this.right.preOrder();
}
}
public Node(int value) {
this.value = value;
}
@Override
public String toString() {
return "Node [value=" + value + "]";
}
@Override
public int compareTo(Node o) {
return this.value - o.value;
}
}
创建赫夫曼树的方法
public static Node createHuffmanTree(int[] arr) {
List<Node> nodes = new ArrayList<Node>();
for (int value : arr) {
nodes.add(new Node(value));
}
while(nodes.size() > 1) {
Collections.sort(nodes);
Node leftNode = nodes.get(0);
Node rightNode = nodes.get(1);
Node parent = new Node(leftNode.value + rightNode.value);
parent.left = leftNode;
parent.right = rightNode;
nodes.remove(leftNode);
nodes.remove(rightNode);
nodes.add(parent);
}
return nodes.get(0);
}
}
最后提供一个遍历输出哈夫曼树的方法
public static void preOrder(Node root) {
if(root != null) {
root.preOrder();
}else{
System.out.println("是空树,不能遍历~~");
}
}