java--赫夫曼树

1.基本介绍

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

2.赫夫曼树创建思路

一个数列{13, 7, 8, 3, 23, 6, 1}，要求转写成一颗赫夫曼树。

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

package com.atguigu.huffmantree;

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, 23, 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<Node>();
        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)从ArryList删除处理过的二叉树
            nodes.remove(leftNode);
            nodes.remove(rightNode);
            //(5)将parent加入到nodes
            nodes.add(parent);

            // System.out.println("第一次处理后" + nodes);
        }
        //返回赫夫曼树的root结点
        return nodes.get(0);
    }
}
    //创建结点类
//为了让node对象持续排序Collection集合排序
//让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;
        }
    }