数据结构讨论课三：二叉树的应用——哈夫曼树的编/译码器的设计和实现

数据结构讨论课三：二叉树的应用——哈夫曼树的编/译码器的设计和实现【皮皮蕾】

---

• 题目要求
• 输出效果
• 源码
• 作者自述

题目要求

在数据通信系统中，电文传送是经常遇到的问题，传送电文时需要将字符转换成二进制组成的字符串，当然在传送电文时，希望传输信息的总长度尽可能的短。这一问题可转化为如何将传送字符集设计一套二进制编码，如何使电文总长最短且不产生二义性。
①读入一个ASCII文件，统计文档中字符出现的频度，构造哈夫曼树；
②在构造好的哈夫曼树中对每个字符进行Huffman编码；
③输出原始数据、每个字符对应的Huffman编码和总编码长度等信息

输出效果

源码

---

解析注释已在代码中标出

#include<iostream>
#include<fstream>
#include<string>
using namespace std;
class node
{
public:
	int weight;//权值
	string ch;//字母
	string code;//哈夫曼编码
	int left,right,parent;
};
void select(node tree[],int *a,int *b,int n)//找权值最小和第二小的a和b
{
	*a=*b=0;
	int min1,min2;
	min1=min2=9999;
	int i;
	for(i=0;i<n;i++)
	{
		if(tree[i].parent==-1)
		{
			if(tree[i].weight<min1)
			{
				min2=min1;
				min1=tree[i].weight;
				*b=*a;
				*a=i;
			}
			else if(tree[i].weight<min2)
			{
				min2=tree[i].weight;
				*b=i;
			}
		}
	}
	///
	/*int temp;
	if(tree[*a].left<tree[*b].left)
	{
		temp=*a;
		*a=*b;
		*b=temp;
	}*/
	///
}
void Huff_tree(node tree[],int w[],string ch[],int n)//构造哈夫曼树
{
	for(int i=0;i<2*n-1;i++)//初始化
	{
		tree[i].parent=-1;
		tree[i].left=-1;
		tree[i].right=-1;
		tree[i].code="";
	}
	for(int i=0;i<n;i++)//赋值
	{
		tree[i].weight=w[i];
		tree[i].ch=ch[i];
	}
	for(int k=n;k<2*n-1;k++)
	{
		int i1,i2;i1=i2=0;
		select(tree,&i1,&i2,k);//将i1，i2结点合成结点k
		tree[i1].parent=k;
		tree[i2].parent=k;
		tree[k].left=i1;
		tree[k].right=i2;
		tree[k].weight=tree[i1].weight+tree[i2].weight;
	}
}
void Huff_code(node tree[],int n)
{
	int i,j,k;string s="";
	for(int i=0;i<n;i++)
	{
		s="";
		j=i;
		while(tree[j].parent!=-1)//从叶子往上找到根节点
		{
			k=tree[j].parent;
			if(j==tree[k].left)//如果j是它父母的左孩子
			{
				s=s+'0';
			}
			else
			{
				s=s+'1';
			}
			j=tree[j].parent;//往上走
		}
		cout<<"字符"<<tree[i].ch<<"的编码：";
		for(int p=s.size()-1;p>=0;p--)
		{
			cout<<s[p];
			tree[i].code+=s[p];//保存编码	
		}
		cout<<endl;
	}
}
class xinxi//储存文件中字符信息:每个字母出现的次数
{
public:
	int number;
	string ch;
};
string str[26];
int w[26];
xinxi s[26];//小写26个英文字母
int sum=0;//文件内小写英文字母的种类
int total=0;//文件内英文字母总数
void tongji()//统计文件字符信息
{
	char word[100];//假定最多100个字符
	//初始化每个字母次数为0//
	for(int i=0;i<26;i++)
	{
		s[i].number=0;
	}
	/读取文本
	ifstream in("data.txt");
	in>>word;//遇到空格停止读入
	for(int i=0;i<strlen(word);i++)
	{
		char m=word[i];
		if(m>='a'&&m<='z')
		{
			total++;
			int n=m-'a';
			s[n].ch=m;
			s[n].number++;
		}
	}
	in.close();
	int hhh=0;
	for(int i=0;i<26;i++)//将s[]中的char信息转移到str[],number信息转移到w[]
	{
		if(s[i].number!=0)
		{
			str[hhh]=s[i].ch;
			w[hhh]=s[i].number;
			hhh++;
			sum++;
		}
	}//str[0~sum]w[0~sum]
}
void WPL(node tree[],int n)
{
	double wpl=0;
	for(int i=0;i<n;i++)
	{
		double x=w[i];
		double y=total;
		double t=x/y;
		double k=tree[i].code.size();
		wpl=wpl+t*k;
	}
	cout<<"WPL="<<wpl<<endl;
}
void codetxt(node tree[],int n)//将哈夫曼编码保存入code.txt文件
{
	//写入文本///
	ofstream out("code.txt");
	for(int i=0;i<n;i++)
	{
		out<<tree[i].code<<endl;
	}
	out.close();
}
void Code(node tree[],int n)//从code.txt中读取哈夫曼编码，转化为字符并存入zhuanhua.txt
{
	ifstream in("code.txt");
	string ee[26];
	for(int i=0;i<n;i++)
		in>>ee[i];
	in.close();
	ofstream out("zhuanhua.txt");
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
		    if(tree[j].code==ee[i])
		        out<<tree[j].ch<<endl;
		}
	}
	out.close();
}
int main()
{
	tongji();//统计data.txt
	node tree[52];
	Huff_tree(tree,w,str,sum);//构造哈夫曼树
	Huff_code(tree,sum);//编码
	WPL(tree,sum);//求WPL
	codetxt(tree,sum);//保存哈夫曼编码至code.txt
	Code(tree,sum);//从code.txt中读取代码，译码后将code.txt中的代码转化为字符，并存入zhuanhua.txt
}

作者自述

写博客实属不易，留个赞再走~——by皮皮蕾