package com.hao.firstdemo.datastruct.tree;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
/**
* @author haoxiansheng
* @data 2020/5/10 15:20
*/
public class HuffmanTree {
public static void main(String[] args) {
int arr[] = {13, 7, 8, 3, 29, 6, 1};
Node root = createHuffmanTree(arr);
preOrder(root);
}
public static void preOrder(Node root) {
if (root != null) {
root.preOrder();
} else {
System.out.println("是空树, 不能遍历");
}
}
/**
* 创建赫夫曼树方法
*
* @param arr 创建赫夫曼树的数组
* @return 返回赫夫曼树的root节点
*/
public static Node createHuffmanTree(int[] arr) {
/**
* 为了方便操作
* 1、遍历数组
* 2、将arr的每个元素构成一个Node
* 3、将Node放入到ArrayList中
*/
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=" + nodes);
// 取出根节点权值最小的两棵树
// (1) 取出权值最小的节点(节点也是二叉树)
Node leftNode = nodes.get(0);
// (2) 取出权值次小的节点(节点也是二叉树)
Node rightNode = nodes.get(1);
// (3) 构建一个新的二叉树
Node parent = new Node(leftNode.value + rightNode.value);
parent.left = leftNode;
parent.right = rightNode;
// (4) 从ArrayList 删除处理过的二叉树
nodes.remove(leftNode);
nodes.remove(rightNode);
// (5) 将parent 加入到nodes中
nodes.add(parent);
}
return nodes.get(0);
}
}
// 创建节点
// 为了让Node 对象支持排序Connections集合排序 Node 实现Comparable接口
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;
}
}