尚硅谷-数据结构与算法-二叉树-哈夫曼树

 

 

package com.atguigu.tree.threadebinarytree;

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

/**
 * 赫夫曼树
 * 所有叶子节点的带权路径长度之和，权值越大离根节点越近的二叉树，WPL最小
 *
 * @author WZ
 * @create 2021-10-24 9:17
 */
public class HuffmanTree {
    public static void main(String[] args) {
        int arr[]=new int[]{13,7,8,3,29,6,1};
        Node root = creareHuffmanTree(arr);
        preSort(root);
    }

    //编写一个前序遍历的方法
    public static void preSort(Node root){
        if (root!=null){
            root.preSort();
        }else {
            System.out.println("该赫夫曼树是空树");
        }

    }

    //创建赫夫曼树的方法

    /**
     *
     * @param arr 需要创建赫夫曼树的数组
     * @return 返回的是创建好后的的赫夫曼树的root节点
     */
    public static Node creareHuffmanTree(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) {//这个主要是因为nodes一直在remove，到最后就剩一个节点。

            //将数据进行排序，从小到大，取决于重写的compareto如何写
            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）从ArrayList中删除处理过的二叉树
            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 preSort(){
        System.out.println(this);
        if (this.left!=null){
            this.left.preSort();
        }
        if (this.right!=null){
            this.right.preSort();
        }
    }



    public Node(int value) {
        this.value = value;
    }

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

    @Override
    public int compareTo(Node o) {
        //this在前，表示从小到大进行排序，如果是从大到小直接-（this.value-o.value）前面添加个负号即可
        return this.value-o.value;//比较权值
    }
}