数据结构与算法——赫夫曼树

1. 基本介绍

1）给定n个权值作为n个叶子节点，构造一棵二叉树，若该树的带权路径长度（wpl）达到最小，称这样的二叉树为最优二叉树，也称为哈夫曼树(Huffman Tree)，还有的书翻译为霍夫曼树。
2）赫夫曼树是带权路径长度最短的树，权值较大的节点离根节点较近。

2.概念介绍

1） 路径和路径长度 ： 在 一 棵 树 中，从一个结点 往下可以达到的孩子或孙子结之间的通路，称为路径。通路中分支的数目称为路径长度。若规定根结点的层数为1，则从根结点到第L层结点的路径长度 L - 1。
2） 结点的权及带权路径长度：若将树中结点赋给一个有着某种含义的数值，则这个数值称为该结点的权。节点的带权路径长度为：从根结点到该结点之间的路径长度与该结点的乘积。
3）树的带权路径长度：树的带权路径长度规定为所有叶子结点的带权路径长度之和，记为WPL(weight path length)，权值越大的结点离根结点越近的二叉树才是最优二叉树。
4）WPL最小的就是赫夫曼树。

3. 创建赫夫曼树的步骤

给你一个数列{13,7,8,3,29,6,1}，要求转为一棵赫夫曼树。

1）从小到大进行排序，将每一个数据，每一个数据都是一个节点，每个节点可以看成是一颗最简单的赫夫曼树
2）取出根节点权值最小的两棵二叉树
3）组成一颗新的二叉树，该新的二叉树的根节点的权值是前面两棵二叉树根结点权值的和
4）再将这颗新的二叉树，以根节点的权值大小再次排序，不断重复1-2-3-4的步骤。直到数列中，所有的数据都被处理，就得到一颗赫夫曼树

package com.sss.huffmantree;

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

public class HuffmanTree {
    public static void main(String[] args) {
        int arr[] = {13,7,8,3,29,6,1};
        Node root = creatHuffmanTree(arr);

        // 测试一把
        root.preOrder();
    }

    // 编写一个前序遍历方法
    public static void preOrder(Node root) {
        if(root != null) {
            root.preOrder();
        } else {
            System.out.println("是空树，不能遍历~~");
        }
    }

    // 创建赫夫曼树的方法
    public static Node creatHuffmanTree(int[] arr) {
        // 第一步为了操作方便
        // 1、遍历arr数组
        // 2、将arr的每一个元素构成一个Node
        // 3、将 Node 放入到ArrayList中
        ArrayList<Node> nodes = new ArrayList<>();
        for (int value : arr) {
            nodes.add(new Node(value));
        }
        // 这是一个循环过程
        while (nodes.size() > 1) {
            // 排序  从小到大
            Collections.sort(nodes);

            // 取出根节点权值最小的两棵二叉树
            // 取出权值最小的结点（二叉树）
            Node leftNode = nodes.get(0);
            // 取出权值第二小的结点（二叉树）
            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);
            // 将parent加入进去
            nodes.add(parent);
        }
        // 返回哈夫曼树的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;
    }

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


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