哈夫曼编码译码器

标题数据结构课设:哈夫曼编码译码器

总结一下这次的课程设计,遇到的问题是如何解决的。
功能:
  1. 将权值数据存放在数据文件data.txt中。
  2. 键盘输入字符集大小n、n个字符和n个权值,建立哈夫曼树。
  3. 利用建好的哈夫曼树生成哈夫曼编码。
  4. 使用data.txt中的权值数据生成的哈夫曼编码,将文件Text.txt中的字符串编码保存到HuffmanCode.txt文件中。
  5. 使用data.txt中的权值数据生成的哈夫曼编码,将文件HuffmanCode.txt中的哈夫曼编码解码生成字符串保存到Text.txt文件中。
  6. 输出哈夫曼编码到code.txt文件。

其中,功能1、2、3、6是基本课程设计要求的功能,为了增加代码的实用性,
导师提议我增加4、5功能。

结构体定义:
typedef struct { //定义二叉树结点类型
    char data ;
	double weight;
    int lchild, rchild ,parent; 
} HTNode , *HuffmanTree;//定义二叉树结点指针类型
typedef struct{//定义哈夫曼编码的结构体
	char cd[11];
	char data;
	int len;
}HCNode , *HuffmanCode;
实现代码:

功能:

  • 建立哈夫曼树哈夫曼编码
  • 键盘输入字符
  • 完成哈夫曼编码和字符的转换
  • 从文件中读取数据并保存
  • 将得到的数据保存到文件中
#include <stdio.h>
#include <stdlib.h>
#include <conio.h>
#include <string.h>
#include "huff2.h"
#define MAX_LINE 1024
//选择最小的两个结点
void Select(HuffmanTree &HT,int i,int &s1,int &s2){
	int k=0;
	s1=i-1;
	s2=i;
	for(int j=i;j>0;j--){
		if(HT[j].parent==0){
			if(HT[j].weight<HT[s2].weight){
				s1=s2;
				s2=j;
			
			}
			if(HT[j].weight<HT[s1].weight&&j!=s2){
				s1=j;
			}
		
		}

	}
 	if(HT[s1].parent!=0){
		s1=1;
		for(;k<i;k++){
			if(HT[k].parent==0){
				if(HT[k].weight<HT[s1].weight){
					s1=k;
				}
			}
		}
	}
}
//构造哈夫曼树及求字符的编码
void HuffmanCoding(HuffmanTree &HT,HuffmanCode &HC,char data[],double weight[],int n){
	int i,m,start,c,f;
	char cd[26]={0};
	if(n<=1) return;
	m = 2*n-1;
	HT = (HuffmanTree)malloc((m+1)*sizeof(HTNode));
	HC = (HuffmanCode)malloc((n+1)*sizeof(HCNode));
	//构建哈夫曼树
	for(i=1;i<=n;i++) {
		HT[i].data = data[i-1];
		HT[i].weight = weight[i-1];
		HT[i].parent = 0;
		HT[i].lchild = 0;
		HT[i].rchild = 0;
	}

	for(;i<=m;i++) {
		HT[i].data = NULL;
		HT[i].weight = 0.000000;
		HT[i].parent = 0;
		HT[i].lchild = 0;
		HT[i].rchild = 0;
	}
	for(i=n+1;i<=m;i++){
		int s1,s2;
		Select(HT,i-1,s1,s2);
		HT[s1].parent = i;
		HT[s2].parent = i;
		HT[i].lchild = s1;
		HT[i].rchild = s2;
		HT[i].weight = HT[s1].weight + HT[s2].weight;
	}
	//求哈夫曼编码
	for(i=1;i<=n;i++){
		start = n-1;
		for(c=i,f=HT[i].parent;f!=0;c=f,f=HT[f].parent){
			if(HT[f].lchild==c)
				cd[--start] = '0';
			else cd[--start] = '1';
		}
		HC[i].len = n-start-1;
		for(int z=0;start<n;start++,z++){
			
			HC[i].cd[z] = cd[start];
			
		}
		HC[i].data = HT[i].data;
	}

}
//将编码表输出到文件code.txt中
void CreateHuffman(HuffmanCode &HC,int n){
	FILE * fp=fopen("code.txt","w+");
	if(fp==NULL){
		printf("无法打开code.txt");
		exit(0);
	}
    for(int i=1;i<=n;i++){
		fputc(HC[i].data,fp);
		fputs("    ",fp);
		fputs(HC[i].cd,fp);
		fputc('\n',fp);
	}
	if(fclose(fp)){
		printf("无法关闭code.txt");
		exit(0);
	}
}

//将编码保存到文件HuffmanCode.txt
void CodePrintTxt(char code[],int i){
	FILE * fp=fopen("HuffmanCode.txt","w+");
	int j;
	if(fp==NULL){
		printf("无法打开HuffmanCode.txt");
		exit(0);
	}
    for(j=0;j<=i;j++){
		fputc(code[j],fp);
	}
	if(fclose(fp)){
		printf("无法关闭HuffmanCode.txt");
		exit(0);
	}
}
//将编码保存到文件Text.txt
void DataPrintTxt(char data[],int i){
	FILE * fp=fopen("Text.txt","w+");
	int j;
	if(fp==NULL){
		printf("无法打开Text.txt");
		exit(0);
	}
    for(j=0;j<=i;j++){
		fputc(data[j],fp);
	}
	if(fclose(fp)){
		printf("无法关闭Text.txt");
		exit(0);
	}
}



//初始化数据之从文件data.txt中读出权值数据
void init1(HuffmanTree &HT,HuffmanCode &HC){
	int j,q,w;
	char data[52]={0};
	double weight[52]={0};

	FILE *file=fopen("data.txt","r");
	if(file==NULL){
		printf("数据为空!!!");
	}else{
		for(j=0;j<182;j++){		
			fscanf(file, "%c%lf", &data[j],&weight[j]);
		}
		fclose(file);
		//将数据正常填装到两个数据集合中
		for(q=0;q<26;q++){
			data[q]=data[2*q];
			weight[q]=weight[2*q];
		}
		for(w=0;w<25;w++){
			for(q=0;q<25;q++){
				if(weight[q]<weight[q+1]){
					char d;
					double w;
					w=weight[q+1];
					weight[q+1]=weight[q];
					weight[q]=w;
					d=data[q+1];
					data[q+1]=data[q];
					data[q]=d;
				}
			}
		}
	}
	HuffmanCoding(HT,HC,data,weight,26);

	CreateHuffman(HC,26);
}
//从键盘输入
void init2(HuffmanTree &HT,HuffmanCode &HC){
	int q,e,i;
	double w;
	char data[26]={0};
	char d[26]={0};
	double weight[26]={0};

	printf("请输入您想定义的字符串\n");
	scanf("%s",data);
	
	//排除相同的字符
	q=0;
	while(data[q]){
		e=q+1;
		while(data[e]){
			if(data[e]==data[q]){
				i=e;	
				while(data[i+1]){
					data[i]=data[i+1];
					i++;
				}
				data[i]=NULL;
			}
			else
				e++;
		}
		q++;
	}
	q=0;
	while(data[q]){
		printf("请输入%c的权值:\n",data[q]);
		scanf("%lf",&w);
		weight[q] = w;
		q++;
	}	
	for(i=0;data[i];i++){
		for(q=0;data[q];q++){
			if(weight[q]<weight[q+1]){
				char d;
				double w;
				w=weight[q+1];
				weight[q+1]=weight[q];
				weight[q]=w;
				d=data[q+1];
				data[q+1]=data[q];
				data[q]=d;
			}
		}
	}
	HuffmanCoding(HT,HC,data,weight,q);
	
	CreateHuffman(HC,q);
}

//将编码转换成数据(解码)
void CodeToData(HuffmanTree HT,char code[]){//n为编码的长度,data为编码
	int i,m,w,c;
	char data[MAX_LINE]={0};
	i=0;c=0;w=0;	
	while(HT[w].parent!=0){
		w++;
	}
	printf("解码为:");
	i=0;
	while(code[i]){
		if(code[i]==' '){
			data[c++] = ' ';
			printf(" ");
		}
		m=w;
		while(1){
			if(code[i]=='0') m=HT[m].lchild;
			else if(code[i]=='1') m = HT[m].rchild;
			else {
				i++;
				if(!code[i])
					break;
				continue;
			}
			i++;
			if(HT[m].lchild==0&&HT[m].rchild==0){
				data[c++] = HT[m].data;
				printf("%c",HT[m].data);
				break;
			}
		}
	}
	printf("\n");
	DataPrintTxt(data,i);//将数据保存到文件Text.txt
}
//将数据转换成编码(编码)
void DataToCode(HuffmanCode HC,char data[]){//data为字符数据,p为保存的编码,n为字符的长度
	int j,k,l,i;
	char code[10000]={0};
	j=0;i=0;
	while(data[j]){
		if(data[j]==' ')
			code[i++] = ' ';
		for(k=1;k<27;k++){
			if(data[j]==HC[k].data){
				//把每个字符的哈夫曼码保存到p中
				for(l=0;l<HC[k].len;l++){
					code[i++] = HC[k].cd[l];
				}
			}
		}
		j++;
	}
	printf("哈夫曼编码为:");
	for(j=0;j<=i;j++){
		printf("%c",code[j]);
	}
	printf("\n");
	CodePrintTxt(code,i);//将编码保存到文件HuffmanCode.txt
}
//读取Text文件内容并保存
void ReadTxtT(HuffmanCode HC,char data[]){//n的作用:用来识别读出哪种文件.
	FILE * fp=fopen("Text.txt","r");
	if(fp==NULL){
		printf("无法打开Text.txt");
		exit(0);
	}

	while(fgets(data,MAX_LINE,fp) != NULL)
	{
		DataToCode(HC,data);//将数据转换成编码
	}

	if(fclose(fp)){
		printf("无法关闭Text.txt");
		exit(0);
	}	
}
//读取HuffmanCode文件内容并保存
void ReadTxtH(HuffmanTree HT,char code[]){//n的作用:用来识别读出哪种文件.
	FILE * fp=fopen("HuffmanCode.txt","r");
	if(fp==NULL){
		printf("无法打开HuffmanCode.txt");
		exit(0);
	}
	while(fgets(code,MAX_LINE,fp) != NULL)
	{
		CodeToData(HT,code);//将编码转换成数据
	}

	if(fclose(fp)){
		printf("无法关闭HuffmanCode.txt");
		exit(0);
	}	
}


二级菜单的显示:

直观的显示出程序的功能

void menu1(){
	printf("		|*******************************************************************************|\n");
	printf("		|-------------------Huffman编码解码器使用菜单-----------------------------------|\n");
	printf("		|-------------------功能1:读取文件并建立哈夫曼树和哈夫曼编码--------------------|\n");
	printf("		|-------------------功能2:键盘输入字符以及权值建立哈夫曼树和哈夫曼编码----------|\n");
	printf("		|-------------------功能3:编码或解码文件中的数据--------------------------------|\n");	
	printf("		|-------------------功能0:退出系统----------------------------------------------|\n");
	printf("		|-------------------请输入数字:0、1、2使用相对应的功能-------------------------|\n");
	printf("		|*******************************************************************************|\n");
}
void menu2(){	
	printf("		|*******************************************************************************|\n");
	printf("		|-------------------Huffman编码解码器使用菜单-----------------------------------|\n");
	printf("		|-------------------功能1:编码--------------------------------------------------|\n");
	printf("		|-------------------功能2:解码--------------------------------------------------|\n");
	printf("		|-------------------功能0:退出系统----------------------------------------------|\n");
	printf("		|-------------------请输入数字:0、1、2使用相对应的功能-------------------------|\n");
	printf("		|*******************************************************************************|\n");
}
主函数小黑窗功能选项的实现:

显示出代码的功能,二级菜单的穿接,友好的提示词。


void main()
{
	int fun;
	char data[MAX_LINE]={0};
	char code[10000]={0};
	HuffmanTree HT; 
	HuffmanCode HC;
	menu1();
	scanf("%d",&fun);
	switch(fun){
		case 1:
			init1(HT,HC);
			menu2();
			scanf("%d",&fun);		
			while(fun){
				switch(fun){
					case 1:
						printf("请输入A-Z字符(大写):\n");
						scanf("%s",data);
						printf("(如果您输入了错误的字符串,则编码为空!)");
						DataToCode(HC,data);
						break;
					case 2:
						printf("请输入编码:\n");
						scanf("%s",code);
						printf("(如果您输入了错误的字符,则跳过错误的字符,不够一个编码的舍去。)");
						CodeToData(HT,code);
						break;
					case 0:
						printf("谢谢使用!\n");
						break;
					default:
						printf("你的输入错误!请重新输入!\n");
						break;
				}
				menu2();
				scanf("%d",&fun);
			}
			
			break;
		case 2:
			init2(HT,HC);
			menu2();
			scanf("%d",&fun);
			while(fun){
			   switch(fun){
					case 1:
						printf("请输入您设定的字符串:\n");
						scanf("%s",data);
						printf("(如果您输入了错误的字符串,则编码为空!)");
						DataToCode(HC,data);
						break;
					case 2:
						printf("请输入编码:\n");
						scanf("%s",code);
						printf("(如果您输入了错误的字符,则跳过错误的字符,不够一个编码的舍去。)");
						CodeToData(HT,code);
						break;
					case 0:
						printf("谢谢使用!\n");
						break;
					default:
						printf("你的输入错误!请重新输入!\n");
						break;
				}
				menu2();
				scanf("%d",&fun);
			}
			break;
		case 3:
			init1(HT,HC);
			menu2();
			scanf("%d",&fun);
			while(fun){
				switch(fun){
					case 1:
						ReadTxtT(HC,data);
						break;
					case 2:
						ReadTxtH(HT,code);
						break;
					case 0:
						printf("谢谢使用!\n");
						break;
					default:
						printf("你的输入错误!请重新输入!\n");
						break;
				}
				menu2();
				scanf("%d",&fun);
			}
			break;
		case 0:
		    printf("谢谢使用!\n");
		    exit(0);
		default:
			printf("你的输入错误!请重新启动!\n");
			break;
	}
}

总结:这次的数据结构使我认识到了,代码应该要认识到代码存在的意义,是为什么服务的,要充分的想到使用者会遇到的误点,通过代码强制阻止或者友好的提示词提示使用者,从而增强代码的健壮性。当实现了最基本的功能之后要看看能不能提高代码的实用性。毕竟只有有用的代码才有存在的意义。最后,bug无处不在,只有一次次的调试、修改才能使bug无处可躲,当遇到暂时无法解决的bug,可以先进行其他的地方,并从别的方向带入,或许就可以解决了。

  • 22
    点赞
  • 104
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值