哈夫曼树

给定n个权值作为n个叶子节点，构造一棵二叉树，若该树的**带权路径长度（wpl）**达到最小，这样的二叉树称为最优二叉树也称为哈夫曼树或霍夫曼树

哈夫曼树是带权路径长度最短的树，权值较大的节点离根较近

package calculate;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class HuffmanTree {

	public static void main(String[] args) {
		int[] arr = {13,7,8,3,29,6,1};
		Node root = createHuffmanTree(arr);
		
		//测试哈夫曼树创建成功
		preOrder(root);

	}
	//编写一个前序遍历的方法
	public static void preOrder(Node root) {
		if (root != null) {
			root.preOrder();
		}else {
			System.out.println("空树不能遍历");
		}
	}
	
	//创建哈夫曼树
	public static Node createHuffmanTree(int[] arr) {
		//1.为了操作方便，遍历arr数组，将arr的每个元素构建成一个Node,将Node放入ArrayList中
		List<Node> nodes = new ArrayList<Node>();
		for (int value:arr) {
			nodes.add(new Node(value));
		}
		//循环过程：当nodes只有一个节点就退出循环
		while(nodes.size()>1) {

			//排序，从小到大
			Collections.sort(nodes);       //因为Node实现了Comparable接口
			System.out.println(nodes);
			//输出：[Node[value=1], Node[value=3], Node[value=6], Node[value=7], Node[value=8], Node[value=13], Node[value=29]]
	
			//取出根节点权值最小的两颗二叉树
			//1.取出权值最小的节点（二叉树）为左子节点
			Node leftNode = nodes.get(0);
			//2.取出第二小的节点（二叉树）为右子节点
			Node rightNode = nodes.get(1);
			//3.构建一棵新的二叉树(父节点:权值之和)
			Node parent = new Node(leftNode.value+rightNode.value);
			//组成一棵树
			parent.left = leftNode;
			parent.right = rightNode;
			//4.从ArrayList删除处理过的二叉树
			nodes.remove(leftNode);
			nodes.remove(rightNode);
			//5.将Parent加入nodes
			nodes.add(parent);
			//System.out.println(nodes);
			//输出：[Node[value=6], Node[value=7], Node[value=8], Node[value=13], Node[value=29], Node[value=4]]
		}
		//返回哈夫曼树的root节点
		return nodes.get(0);
	}

}

//创建节点类
//为了让Node对象支持排序Collections集合排序
//让Node实现Compareble接口
class Node implements Comparable<Node>{
	int value; //节点权值
	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() {
		// TODO Auto-generated method stub
		return "Node[value="+value+"]";
	}

	@Override
	public int compareTo(Node o) {
		// TODO Auto-generated method stub
		return this.value-o.value;  //从小到大排序
		//若要从大到小排序就是-(this.value-o.value)
	}
	
	
}