哈夫曼树介绍及Java实现

1. 介绍

1.1 哈夫曼树

  • 哈夫曼树-最优二叉树:树的带权路径长度最小的二叉树;
  • 权值均为叶子结点;
  • 权值越大,结点距离根节点越近;

在这里插入图片描述

1.2 路径、路径长度、结点的权、结点的带权路径长度

在这里插入图片描述

1.3 树的带权路径长度WPL

在这里插入图片描述

2. 哈夫曼树构建步骤

  • 对权值序列进行升序排列,选出其中最小的两个权值进行合并;
  • 将合并结点加入权值序列,重新进行排序;
  • 重复执行上述两个操作,直到序列中只剩一个节点即可完成创建;
    在这里插入图片描述
  • 上述案例的手动构建结果:
    在这里插入图片描述

3. 代码实现

package com.northsmile.tree.huffman;

import java.util.*;

/**
 * @author NorthSmile
 * @version 1.0
 * @date 2023/8/21&18:36
 * 哈夫曼树实现
 */
public class HuffmanTree {
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        int n=Integer.parseInt(scanner.nextLine());
        List<TreeNode> nodes=new LinkedList<>();
        for (int i=0;i<n;i++){
            nodes.add(new TreeNode(Integer.parseInt(scanner.nextLine())));
        }
        root=createHuffmanTree(nodes);
        // System.out.println(root.val);
        System.out.println(calWPL());
    }
    // 根节点
    private static TreeNode root;

    // 构建哈夫曼树
    public static TreeNode createHuffmanTree(List<TreeNode> nodes){
        while (nodes.size()!=1){
            // 升序排列
            nodes.sort(new Comparator<TreeNode>() {
                @Override
                public int compare(TreeNode o1, TreeNode o2) {
                    return o1.val-o2.val;
                }
            });
            // 合并两个最小节点
            TreeNode left=nodes.get(0);
            TreeNode right=nodes.get(1);
            // System.out.println(left.val+" "+right.val);
            TreeNode curRoot = new TreeNode(left.val+right.val);
            curRoot.left=left;
            curRoot.right=right;
            nodes.add(curRoot);
            // 删除两个最小节点
            nodes.remove(0);
            nodes.remove(0);
        }
        return nodes.get(0);
    }

    // 计算WPL
    // 层序遍历
    public static int calWPL(){
        int wpl=0;
        Deque<TreeNode> queue=new ArrayDeque<>();
        queue.offer(root);
        int height=0;
        while (!queue.isEmpty()){
            int size=queue.size();
            for (int i=0;i<size;i++){
                TreeNode cur = queue.pop();
                if (cur.left==null&&cur.right==null){
                    wpl+=(cur.val*height);
                }
                if (cur.left!=null){
                    queue.offer(cur.left);
                }
                if (cur.right!=null){
                    queue.offer(cur.right);
                }
            }
            height++;
        }
        return wpl;
    }

    // 树节点
    private static class TreeNode{
        public int val;
        public TreeNode left;
        public TreeNode right;

        public TreeNode(int val){
            this.val=val;
        }
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
哈夫曼树是一种带权路径长度最短的树,常用于数据压缩。在Java中,可以通过构建哈夫曼树实现哈夫曼编码。下面是一个简单的哈夫曼树Java实现的步骤: 1. 定义一个Node类作为哈夫曼树的节点,包含权值和左右子节点等属性。 2. 构建哈夫曼树,可以通过优先队列(PriorityQueue)来实现。首先将所有节点加入队列中,然后每次取出权值最小的两个节点,合并成一个新节点,再将新节点加入队列中,直到队列中只剩下一个节点,即为哈夫曼树的根节点。 3. 根据哈夫曼树生成哈夫曼编码,可以通过递归遍历哈夫曼树实现。从根节点开始遍历,每次向左走为0,向右走为1,直到叶子节点,即可得到该节点对应的哈夫曼编码。 下面是一个简单的Java代码实现: ```java public class HuffmanTree { private Node root; private class Node implements Comparable<Node> { int weight; Node left; Node right; public Node(int weight) { this.weight = weight; } public int compareTo(Node other) { return this.weight - other.weight; } public boolean isLeaf() { return left == null && right == null; } public String toString() { return "Node(" + weight + ")"; } } public HuffmanTree(int[] weights) { PriorityQueue<Node> pq = new PriorityQueue<>(); for (int weight : weights) { pq.offer(new Node(weight)); } while (pq.size() > 1) { Node left = pq.poll(); Node right = pq.poll(); Node parent = new Node(left.weight + right.weight); parent.left = left; parent.right = right; pq.offer(parent); } root = pq.poll(); } public Map<Character, String> getCodes() { Map<Character, String> codes = new HashMap<>(); getCode(root, "", codes); return codes; } private void getCode(Node node, String code, Map<Character, String> codes) { if (node.isLeaf()) { codes.put((char) node.weight, code); } else { getCode(node.left, code + "0", codes); getCode(node.right, code + "1", codes); } } } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

NorthSmile

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值