哈夫曼树(java实现)

package com.fcy.dataStruct;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;
public class HuffmanTree {
	//节点类
	public static class Node<E>{
		E data;
		double weight;       //权重
		Node<E> leftChild;
		Node<E> rightChild;
		public Node(E data,double weight){
			this.data=data;
			this.weight=weight;
		}
		public String toString(){
			return "Node[data="+data+",weight="+weight+"]";
		}
	}
	private static<E> Node<E> createTree(List<Node<E>> nodes){
		//只要nodes数组中有两个以上的节点
		while(nodes.size()>1){
			quickSort(nodes);
			//获取权值最小的两个节点
			Node<E> left=nodes.get(nodes.size()-1);
			Node<E> right=nodes.get(nodes.size()-2);
			//生成新节点,新节点权值为两个子节点之和
		Node<E> parent=new Node<>(null,left.weight+right.weight);
			//让新节点作为权值最小的两个节点的父节点
			parent.leftChild=left;
			parent.rightChild=right;
			//删除权值最小的两个节点
			nodes.remove(nodes.size()-1);
			nodes.remove(nodes.size()-1);
			//将新生成的父节点添加到集合中
			nodes.add(parent);
		}
		return nodes.get(0);
	}
	private static<E> void swap(List<Node<E>> nodes,int i,int j){
		Node<E> tmp;
		tmp=nodes.get(i);
		nodes.set(i,nodes.get(j));
		nodes.set(j,tmp);
	}
	//实现快速排序算法,用于对节点进行排序(用其他的排序算法也可)
	private static <E> void subSort(List<Node<E>> 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);
		}
	}
	public static<E> void quickSort(List<Node<E>> nodes){
		subSort(nodes,0,nodes.size()-1);
	}
	public static List<Node> breadthFirst(Node root){
		Queue<Node> queue=new ArrayDeque<Node>();
		List<Node> list=new ArrayList<Node>();
		if(root!=null){
			queue.offer(root);
		}
		while(!queue.isEmpty()){
			list.add(queue.peek());
			Node p=queue.poll();
			if(p.leftChild!=null){
				queue.offer(p.leftChild);
			}
			if(p.rightChild!=null){
				queue.offer(p.rightChild);
			}
		}
		return list;
	}
	public static void main(String[] args) {
		List<Node<String>> nodes=new ArrayList<>();
		nodes.add(new Node<String>("A",7.0));
		nodes.add(new Node<String>("B",5.0));
		nodes.add(new Node<String>("C",2.0));
		nodes.add(new Node<String>("D",4.0));
		Node<String> root=HuffmanTree.createTree(nodes);
		List<Node> list=breadthFirst(root);
		for(Node l:list){
			System.out.println(l);
		}
	}
}

 


运行结果:




  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值