给定N个权值作为N个叶子结点,构造一棵二叉树,若该树的带权路径长度达到最小,称这样的二叉树为最优二叉树,也称为哈夫曼树(Huffman Tree)。哈夫曼树是带权路径长度最短的树,权值较大的结点离根较近。
构建哈夫曼树的过程:
1、首先对数组中的数字进行排序,按照从小到大排序;
2、排好序之后,将这些数字放入顺序存储结构中,选择最小的两个数字,分别以这两个数为左子节点与右子节点,以这两个数字之和为根节点,构造成一个新的二叉树;
3、在存储结构中删除这两个子节点,并将新的根节点放入存储结构中,重新排序;
4、按照上面的步骤依次循环,直到只剩下一个节点。
import java.util.*;
public class Main {
public static void main(String[] args) {
int[] arr = {10, 4, 5, 6, 23, 44};
Node root = createHuffmanTree(arr);
System.out.println("哈夫曼树前序遍历结果为:");
preOrder(root);
}
//哈夫曼树方法
public static Node createHuffmanTree(int[] arr) {
//遍历数组,并将数组每个元素变为Node类型,并且放入顺序表中
List<Node> nodes = new ArrayList<Node>();
for (int i : arr) {
nodes.add(new Node(i));
}
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("该二叉树为空");
}
}
}
//实现Comparable接口
class Node implements Comparable<Node> {
int value; //权值大小
Node left;
Node right;
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 void preOrder() {
System.out.println(this);
if (this.left != null) {
this.left.preOrder();
}
if (this.right != null) {
this.right.preOrder();
}
}
}
哈夫曼树前序遍历结果为:
Node{value=92}
Node{value=44}
Node{value=48}
Node{value=23}
Node{value=25}
Node{value=10}
Node{value=15}
Node{value=6}
Node{value=9}
Node{value=4}
Node{value=5}