一、介绍
二、说明
三、图集
四、构建步骤
五、代码
package cn.qqqking.huffmantree;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
/**
* @author AnQi
* @date 2020/4/25 09 58:28
* @description
*/
public class HuffmanTree {
public static void main(String[] args) {
int arr[]={13,7,8,3,29,6,1};
Node huffmanTree = createHuffmanTree(arr);
preOrder(huffmanTree);
}
//前序
public static void preOrder(Node root){
if(root != null){
root.preOrder();
}else{
System.out.println("空!!!");
}
}
//创建哈夫曼树方法
public static Node createHuffmanTree(int [] arr){
//第一步为了操作方便
//1、遍历arr数组
//2、将arr每个元素构建成node
//3、将node放入 ArrList 中
List<Node> nodes = new ArrayList<>();
for (int value : arr) {
nodes.add(new Node(value));
}
while(nodes.size()>1){
//排序
Collections.sort(nodes);
System.out.println(nodes);
//取出根节点最小的两颗二叉树
//(1)取出权值最小的节点(2叉树)
Node leftNode = nodes.get(0);
//(2)取出权值第二小的节点(2叉树)
Node rightNode = nodes.get(1);
//(3)构建新的二叉树
Node parent = new Node(leftNode.value+rightNode.value);
parent.left = leftNode;
parent.right = rightNode;
//(4) ArrList 干掉 0 1
nodes.remove(leftNode);
nodes.remove(rightNode);
//(5) parent 加入 ArrList
nodes.add(parent);
}
//返回root节点
return nodes.get(0);
}
}
//创建节点类
//为了让node对象 支持排序
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();
}
}
@Override
public int compareTo(Node o) {
return this.value-o.value;
}
public Node(int value) {
this.value = value;
}
@Override
public String toString() {
return "Node{" +
"value=" + value +
'}';
}
}