思路
- 从小到大进行排序, 将每一个数据,每个数据都是一个节点 , 每个节点可以看成是一颗最简单的
- 取出根节点权值最小的两颗二叉树
- 组成一颗新的二叉树, 该新的二叉树的根节点的权值是前面两颗二叉树根节点权值的和
- 再将这颗新的二叉树,以根节点的权值大小 再次排序, 不断重复 1-2-3-4 的步骤,直到数列中,所有的数据都被处理,就得到一颗赫曼树
代码
1、创建节点
class NodeHaffman implements Comparable<NodeHaffman>{
int value;
NodeHaffman left;
NodeHaffman right;
public NodeHaffman(int value) {
this.value = value;
}
@Override
public int compareTo(NodeHaffman o) {
return this.value-o.value;
}
//前序遍历
public void pre(NodeHaffman node) {
if(node==null) {
return;
}
System.out.println(node.value + " ");
pre(node.left);
pre(node.right);
}
}
2、实现哈夫曼树的创建
class HaffmanCreate {
//创建哈夫曼树
public static NodeHaffman create(int[] arr) {
List<NodeHaffman> list = new ArrayList<>();
for(int i=0; i<arr.length; i++) {
list.add(new NodeHaffman(arr[i]));
}
while(list.size() > 1) { //遍历列表
//对列表进行排序
Collections.sort(list);
//取出前两个值
NodeHaffman temp1 = list.get(0);
NodeHaffman temp2 = list.get(1);
//将两个数进行相加得到父节点
NodeHaffman parent = new NodeHaffman(temp1.value + temp2.value);
parent.left = temp1;
parent.right = temp2;
//从列表中删除最小的两个节点
list.remove(temp1);
list.remove(temp2);
//将新节点加入到列表中
list.add(parent);
}
return list.get(0);
}
}
3、测试
public class HaffmanTree {
public static void main(String[] args) {
int[] arr = {13, 7, 8, 3, 29, 6, 1};
NodeHaffman node = new HaffmanCreate().create(arr);
node.pre(node);
}
}