哈夫曼树实现并得到哈夫曼编码

哈夫曼编码的详细内容可以看这个大佬:哈夫曼(huffman)树和哈夫曼编码

#include <bits/stdc++.h>

using namespace std;

typedef struct Node* node;

struct Node{
	int value;
	char tx;
	node left,right;
	Node(){
		value = 0;
		left = right = NULL;
	}
};

struct cmp{
	bool operator()(node a,node b){
		return a->value > b->value;
	}
};

priority_queue<node,vector<node>,cmp> Q;

string text;

map<char,int> book;

inline void CheckOne(){
	map<char,int>::iterator it;
	int num = 0;
	for(it=book.begin() ; it!=book.end() ; ++it){
		printf("%d: %c---%d\n",++num,(*it).first,(*it).second);
	}
}

node Solved(node a,node b){
	node c = new Node();
	c->value = a->value + b->value;
	c->left = a;
	c->right = b;
	return c;
}

inline void Q_Push(){
	map<char,int>::iterator it;
	for(it=book.begin() ; it!=book.end() ; ++it){
		node p = new Node();
		p->value = (*it).second;
		p->tx = (*it).first;
		Q.push(p);
	}
}

void getCode(node rt,string code){
	if(rt->left == NULL && rt->right == NULL){
		cout << rt->tx << ": " << code <<endl;
		return ;
	}
	getCode(rt->left,code+'0');
	getCode(rt->right,code+'1');
}

inline void init(){
	book.clear();
	while(!Q.empty())Q.pop();
}

int main(){
	
	init();
	cin >> text;//读入文本串 
	for(int i=0 ; i<text.length() ; ++i){
		char ch = text[i];
		if(book.find(ch) != book.end())++book[ch];
		else book.insert(make_pair(ch,1));
	}
	//CheckOne();//检查各字符权值 
	Q_Push();//将各字符权值数据封装入结构体并将指针push 
	node A,B;
	while(Q.size() > 1){
		A = Q.top();
		Q.pop();
		B = Q.top();
		Q.pop();
		Q.push(Solved(A,B)); 
	}
	node rt = Q.top();
	Q.pop();
	getCode(rt,"");//得到各字符对应的HuffmanCode 
	
	return 0;
}

 

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值