图解

代码实现

package com.atguigu.huffmanTree;import javax.swing.*;
import java.awt.geom.RoundRectangle2D;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;/*** @创建人 wdl* @创建时间 2021/3/26* @描述*/
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.遍历arr数组//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);//取出根节点权值最小的两颗二叉树//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加入到nodesnodes.add(parent);//            System.out.println("第一次处理后"+nodes);}//返回赫夫曼树的root节点return nodes.get(0);}}
//创建节点类
//为了让Node对象支持排序Collections集合排序
//让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;}@Overridepublic String toString() {return "Node{" +"value=" + value +'}';}@Overridepublic int compareTo(Node o) {//从小到大进行排序return this.value-o.value;}
}