package tree;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
/**
* @author taoke
* @desc 赫夫曼树
* @email 1504806660@qq.com
* @date 2022/1/12
*/
public class HuffmanTree {
public static void main(String[] args) {
int[] arr = {13, 7, 8, 3, 29, 6, 1};
List<Node> list = new ArrayList<>();
for (int i : arr) {
list.add(new Node(i));
}
Node node = createHuffmanTree(list);
node.preOrder();
}
/**
* 链表
*/
public static class Node implements Comparable<Node> {
public Node left;
public Node right;
public int val;
public Node(int val) {
this.val = val;
}
@Override
public String toString() {
return "Node{" +
"val=" + val +
'}';
}
/**
* 前序遍历
*/
public void preOrder() {
System.out.println(this);
if (this.left != null) {
this.left.preOrder();
}
if (this.right != null) {
this.right.preOrder();
}
}
@Override
public int compareTo(Node o) {
return this.val - o.val;
}
}
/**
* 创建赫夫曼树
* 1、从小到大进行排序,每个数据都是一个节点,每个节点都可以看出是一颗最简单的二叉树
* 2、去除根节点权值最小的两颗二叉树
* 3、组成一颗新的二叉树,该新的二叉树的根节点的权值是前面两颗二叉树节点权值的和
* 4、再将这颗新的二叉树,以根节点的权值大小再次排序,不断重复1,2,3,4的步骤,直到所有的数据都被处理,就得到一颗赫夫曼树
*
* @param list node节点列表
* @return 赫夫曼树
*/
public static Node createHuffmanTree(List<Node> list) {
//当list只有一个元素代表赫夫曼树已经创建好了
//每次将节点排好序,取前两个构成新的节点,移除原来的前两个节点,
//将新的节点添加进list。最终只剩下一颗赫夫曼树
while (list.size() > 1) {
Collections.sort(list);
Node leftNode = list.get(0);
Node rightNode = list.get(1);
Node parent = new Node(leftNode.val + rightNode.val);
parent.left = leftNode;
parent.right = rightNode;
list.remove(leftNode);
list.remove(rightNode);
list.add(parent);
}
return list.get(0);
}
}
数据结构与算法-树-赫夫曼树
于 2022-01-12 15:33:54 首次发布