哈夫曼树(Java)

带权路径最小的树被称为哈夫曼树或者最优二叉树;

n个叶子节点的哈夫曼树总共有2*n-1个节点;

package com.answer.binaryTree;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;

public class HuffmanTree {
    private static class Node<T>{
        private String data;
        private double weight;
        private Node left;
        private Node right;
        public Node(String data,double weight){
            this.data=data;
            this.weight=weight;
        }
        public String toString(){
            return "data:"+this.data+";weight:"+this.weight;
        }
    }
    private Node createTree(List<Node> nodes){
        while(nodes.size()>1){
            quickSort(nodes);
            Node left=nodes.get(nodes.size()-1);
            Node right=nodes.get(nodes.size()-2);
            Node parent=new Node(null,left.weight+right.weight);
            parent.left=left;
            parent.right=right;
            nodes.remove(left);
            nodes.remove(right);
            nodes.add(parent);
        }return nodes.get(0);
    }

    private void quickSort(List<Node> nodes) {
        subSort(nodes,0,nodes.size()-1);
    }

    private void subSort(List<Node> nodes, int start, int end) {
        if(start<end){
            Node base=nodes.get(start);
            int i=start;
            int j=end+1;
            while(true){
                while (i<end&&nodes.get(++i).weight>=base.weight);
                while(j>start&&nodes.get(--j).weight<=base.weight);
                if(i<j){
                    swap(nodes,i,j);
                }else {
                    break;
                }
            }
            swap(nodes,start,j);
            subSort(nodes,start,j-1);
            subSort(nodes,j+1,end);
        }

    }
    private void swap(List<Node> nodes,int i,int j){
        Node temp=nodes.get(i);
        nodes.set(i,nodes.get(j));
        nodes.set(j,temp);
    }
    public ArrayList<Node> level(Node root){
        ArrayList<Node> list=new ArrayList<>();
        Queue<Node> queue=new ArrayDeque<>();
        if(root!=null){
            queue.add(root);
        }
        while (queue.isEmpty()==false){
            Node node1=queue.remove();
            list.add(node1);
            if(node1.left!=null){
                queue.add(node1.left);
            }
            if(node1.right!=null){
                queue.add(node1.right);
            }
        }return list;
    }

    public static void main(String[] args) {
        List<Node> nodes = new ArrayList<>();
        nodes.add(new Node<String>("A" , 40.0));
        nodes.add(new Node<String>("B" , 7.0));
        nodes.add(new Node<String>("C" , 10.0));
        nodes.add(new Node<String>("D" , 30.0));
        nodes.add(new Node<String>("E" , 12.0));
        nodes.add(new Node<String>("F" , 2.0));
        HuffmanTree tree=new HuffmanTree();
        Node root=tree.createTree(nodes);
        /*ArrayList<Node> list=tree.level(root);
        list.forEach(node -> System.out.println(node.weight));*/
        System.out.println(tree.level(root));

    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 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
发出的红包

打赏作者

Dream答案

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

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

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

打赏作者

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

抵扣说明:

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

余额充值