数据结构之哈弗曼树

一、引入
问：数据结构干什么的？
答：存储数据
所以，学数据结构这一部分都是为了存储数据服务的，这里先总结哈弗曼树
二哈弗曼树


1.基本概念： 哈夫曼树即最优二叉树，是只有叶节点才能存放数据，且使所有数据到根节点的距离与其权值乘积和最小的二叉树。

2.建立哈夫曼树的步骤：
1.从所有数据中找到权值最小的两个数据A和B,把A和B作为两个子节点，建立
一个二叉树，设它们的父节点为C，C的权值是A和B权值的和。
2.在原来的所有数据中用C替换A和B，重复上一个步骤
3.直到只剩下一个数据时哈夫曼树完成。

代码如下

public class HFMTree {

	public HTreeNode createTree(int[] data) {
		HTreeNode[] nodes = new HTreeNode[data.length];
		// 把数字数组转换为结点数组
		for (int i = 0; i < data.length; i++) {
			HTreeNode node = new HTreeNode();
			node.data = data[i];
			nodes[i] = node;
		}
		while (nodes.length > 1) {
			sort(nodes);
			// 找出两个最小的结点
			HTreeNode n1 = nodes[0];
			HTreeNode n2 = nodes[1];
			// 把两个最小的结点组成一个新的结点
			HTreeNode n3 = new HTreeNode();
			n3.data = n1.data + n2.data;
			n3.left = n1;
			n3.right = n2;
			// 新建一个数组，把原来最小的两个结点去掉，把新结点加进去
			HTreeNode[] nodes2 = new HTreeNode[nodes.length - 1];
			for (int i = 0; i < nodes2.length - 1; i++) {
				nodes2[i] = nodes[i + 2];
			}
			// 把新建结点加到新数组的最后
			nodes2[nodes2.length - 1] = n3;
			// 用新数组代替原来的数组
			nodes = nodes2;
		}

		return nodes[0];
	}

	// 输出霍夫曼编码
	public void print(HTreeNode node, String code) {
		if (node.left == null && node.right == null) {
			System.out.println(node.data + "的霍夫曼编码是：" + code);
		}
		if (node.left != null) {
			print(node.left, code + "0");
		}
		if (node.right != null) {
			print(node.right, code + "1");
		}
	}

	// 对结点数组用冒泡排序法排序
	public void sort(HTreeNode[] nodes) {
		for (int i = 0; i < nodes.length; i++) {
			for (int j = i+1; j <nodes.length; j++) {
				// 交换结点的位置而不是结点的数据
				if (nodes[i].data > nodes[j].data) {
					HTreeNode node = nodes[i];
					nodes[i] = nodes[j];
					nodes[j] = node;
				}
			}
		}
	}

	public static void main(String[] args) {
		HFMTree hf = new HFMTree();
		int[] data = { 1, 3, 6, 19, 7, 10, 15 };
		HTreeNode rootNode = hf.createTree(data);
		hf.print(rootNode, "");
	}
}