非线性数据结构之赫夫曼树

基本介绍

特点

{13,7,8,3,29,6,1} 的赫夫曼树为:

案例

package com.example.dataalgorithm.tree.huggmantree;

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

/**
 * 哈夫曼树
 *
 * @author qb
 * @version 1.0
 * @since 2022/3/3 10:50
 */
public class HuffmanTree {

    public static void main(String[] args) {

        int[] arr = {13,7,8,3,29,6,1};
        Node resNode = createHuffmanTree(arr);
        //67-29-38-15-7-8-23-10-4-1-3-6-13
        preOrder(resNode);



    }

    public static void preOrder(Node root){
        if(root != null){
            root.preOrder();
        }else {
            System.out.println("空树不能遍历");
        }
    }

    public static Node createHuffmanTree(int[] arr){
        List<Node> nodes = new ArrayList<>();
        //1.为了操作方便遍历arr 数组
        //2.将arr的每个元素构建成一个node
        //3.将node放入到arrayList中
        for (int value: arr){
            nodes.add(new Node(value));
        }

        while (nodes.size() > 1){
            //处理过程时循环的过程
            //排序  从小到大
            Collections.sort(nodes);
            //取出根节点权值最小的两颗二叉树
            //(1) 取出最小的二叉树
            Node left = nodes.get(0);
            //(2) 取出第二小的二叉树
            Node right = nodes.get(1);
            //(3) 构建成新的二叉树
            Node parent = new Node(left.value+right.value);
            parent.left = left;
            parent.right = right;
            //(4) 从arrayList中删除处理过的二叉树
            nodes.remove(left);
            nodes.remove(right);
            nodes.add(parent);
        }
        //返回最后节点，赫夫曼树的 root节点
        return nodes.get(0);
    }

}

/**
 * 为了让node对象 支持排序Collections集合排序
 */
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) {
        return this.value - o.value;
    }

    /**
     * 前续遍历
     */
    public void preOrder(){

        System.out.println(this);
        if(null != this.left){
            this.left.preOrder();
        }
        if(null != this.right){
            this.right.preOrder();
        }
    }
}