给定N个权值作为N个叶子节点,构造一棵二叉树,若该树的带权路径长度达到最小,称这样的二叉树为最优二叉树,也称为赫夫曼树
赫夫曼树是带权路径长度最短的树,权值较大的结点离根较近
节点的带权路径长度为:从根节点到该节点之间的路径长度与该节点的权的乘机
树的带权路径长度(WPL):所有叶子节点的带权路径长度之和
WPL最小的就是赫夫曼树
步骤:从小到大排序后,取出前2个数据,组成一个二叉树,其和为这两个数据的父节点,将该父节点的值加入数据再次排序,再取出前2个数据,与刚才之和组合,其和为这两个数据的父节点,重复直到所有数据取出,赫夫曼树创建完成
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};
preOrder(createHuffmanTree(arr));
}
public static void preOrder(Node root){
if (root!=null){
root.preOrder();
}
}
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);
}
}
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() {
return "Node[" +
"value=" + value +
']';
}
@Override
public int compareTo(Node o) {
return this.value - o.value;
}
}
来源:https://www.cnblogs.com/bingbug/p/12298523.html