22. 赫夫曼树的讲解，以及JAVA代码如何创建赫夫曼树

1. 什么是赫夫曼树

• 给定N个权值作为N个叶子结点，构造一棵二叉树，若该树的带权路径长度达到最小，称这样的二叉树为最优二叉树，也称为哈夫曼树(Huffman Tree)。哈夫曼树是带权路径长度最短的树，权值较大的结点离根较近
• 哈夫曼树又称最优二叉树，是一种带权路径长度最短的二叉树。所谓树的带权路径长度，就是树中所有的叶结点的权值乘上其到根结点的路径长度（若根结点为0层，叶结点到根结点的路径长度为叶结点的层数）。树的路径长度是从树根到每一结点的路径长度之和，记为WPL=（W1L1+W2L2+W3L3+…+WnLn），N个权值Wi（i=1,2,…n）构成一棵有N个叶结点的二叉树，相应的叶结点的路径长度为Li（i=1,2,…n）。可以证明哈夫曼树的WPL是最小的。

2. 代码实现创建赫夫曼树

• 代码实现

package com.qin.tree;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

//赫夫曼树
public class HuffmanTree {

    public static void main(String[] args) {

        int[] arr = {13,7,8,3,29,6,1};
        Node huffmanTree = createHuffmanTree(arr);
        System.out.println("=============");
        //测试 使用前序遍历
        preOrder(huffmanTree);
    }

    //创建赫夫曼树的方法
    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=" + nodes);

            //取出跟节点权值最小的两颗二叉树
            // 1. 取出权值最小的节点（二叉树）
            Node leftNode = nodes.get(0);
            // 2. 取出权值次小的节点（二叉树）
            Node rightNode = nodes.get(1);

            //构建一颗新的二叉树,左节点的权值+有节点的权值 = 父节点的权值
            Node parent = new Node(leftNode.value+rightNode.value);
            parent.left = leftNode;
            parent.right = rightNode;

            //由于现在已经构建了一个新的二叉树，所以我们需要把原来在arrayList中的两个数据删掉
            nodes.remove(leftNode);
            nodes.remove(rightNode);

            nodes.add(parent);

        }
        //返回赫夫曼树的root节点
        return nodes.get(0);
    }

    //写一个前序遍历的方法
    public static void preOrder(Node root){
        if (root!=null){
            root.preOrder();
        }else {
            System.out.println("二叉树为空，无法遍历");
        }
    }



}

//创建节点类
//为了让Node对象，支持排序Collection集合排序
//让Node实现Comparable接口
class Node implements Comparable<Node>{

    int value;//节点的权值
    Node left;//左指针
    Node right;//右指针

    //有参构造
    public Node(int value) {
        this.value = value;
    }

    @Override
    public String toString() {
        return "Node{" +
                "value=" + value +
                '}';
    }


    @Override
    public int compareTo(Node o) {
        //-(this.value - o.value) 从大到小排序
        return this.value - o.value ;//表示从小到大进行排序
    }

    //前序遍历
    public void preOrder(){
        System.out.println(this);
        if (this.left!=null){
            this.left.preOrder();
        }
        if (this.right!=null){
            this.right.preOrder();
        }
    }


}