标题数据结构课设:哈夫曼编码译码器
总结一下这次的课程设计,遇到的问题是如何解决的。
功能:
- 将权值数据存放在数据文件data.txt中。
- 键盘输入字符集大小n、n个字符和n个权值,建立哈夫曼树。
- 利用建好的哈夫曼树生成哈夫曼编码。
- 使用data.txt中的权值数据生成的哈夫曼编码,将文件Text.txt中的字符串编码保存到HuffmanCode.txt文件中。
- 使用data.txt中的权值数据生成的哈夫曼编码,将文件HuffmanCode.txt中的哈夫曼编码解码生成字符串保存到Text.txt文件中。
- 输出哈夫曼编码到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,可以先进行其他的地方,并从别的方向带入,或许就可以解决了。