成长路上的小程序之—— 哈夫曼编码、译码

这是大二数据结构第七次上机老师布置的任务：实现文件操作，对文件进行哈夫曼编码、译码

之所以为此写一篇博客，是因为自认为这个程序对我的意义比较重大。

我是以一个课程设计的要求来写的，大一结束的暑假也做了一个课程设计：《学生通讯录》

但是太水了，完全没有难度。

这个相对来说则有一些巧妙的思想，完全是我独立完成的！

哈哈哈

代码如下：

#include <cstdlib>
#include <iostream>
#include <string.h>
#include <ctype.h>
#include <conio.h>
#include "huffmantree.h"
#define MAX 1000000
using namespace std;

char str[MAX],ch[20];
int  n=0;
FILE *fp1, *fp2;
HuffmanTree HT;
HuffmanCode HC;

int main(int argc, char *argv[])
{
    void menu(void);
    menu();
    system("PAUSE");
    return EXIT_SUCCESS;
}

void menu(void)//菜单 
{
    void encoded(void);//编码 
    void decoded(void);//译码 
    system("cls");
    printf("\n\t\t**************编码译码菜单********\n\n");
    printf("\n\t\t**************1、编码\n");
    printf("\n\t\t**************2、译码\n");
    printf("\n\t\t**************3、退出\n");
    printf("\n\t\t**********************************\n\n");
    printf("\n\n\t\t************请选择(1/2/3)：\n\n");
    char ch=getch();//不需要回车 
    switch(ch)
    { 
     case '1': encoded();break;
     case '2': decoded();break;
     case '3': exit(0);
    }
    
}

void encoded()
{
    int w[30];
    printf("\n\t\t******************编码过程如下：\n"); 
    //编码 
    fp1 = fopen("in.txt", "r");
    fp2 = fopen("out.txt", "w");
    fgets(&str[1],MAX,fp1);
    //文件读取完成 ，从1下标处开始读取，因为哈夫曼编码从1开始 
    //这里&str[1]个人觉得比较巧妙  ^_^ 
    int len = strlen(&str[1]);//字符总个数 
    printf("\n\t\t******************字符总个数为 %d 个\n",len); 
    int count[30] = {0};
    for(int i=1; i<=len; i++)
     if(isalpha(str[i]))
     {
      str[i] = tolower(str[i]);//全部转换为小写字母 
      count[str[i]-'a'+1]++;//统计字符出现个数，简单明了，值得表扬 ^_^ 
     }
    for (int i=1; i<=26; i++)
     if (count[i])
     {
      w[i] = count[i];//赋给哈夫曼中的权值 
      n++;//统计出现字母类别个数 
     }
    HuffmanCoding(HT,HC,w,n);//调用哈夫曼 
    
    for(int i=1; i<len; i++)
    {
     if(isalpha(str[i]))
      fputs(HC[str[i]-'a'+1],fp2);
     else
      fprintf(fp2,"%c",str[i]);
    }   
    fclose(fp1);
    fclose(fp2);
    printf("\n\n\t\t编码成功！\n");
    printf("\n\n\t\t编码结果保存在out.txt文件中");
    printf("请查看文件! ");
    system("pause"); 
    menu();
}
    
void decoded()
{
    void ncpy(char ch[],char* str,int index,FILE* fp,int* countc);
    printf("\n\t\t******************译码过程如下：\n");
    fp1=fopen("out.txt","r");
    fp2=fopen("result.txt","w");
    fgets(str,MAX,fp1);//取出编码结果 
    int index=2;
    int len=strlen(str);
    int countc=1;
    for(int j=0; j<len; )
    {
     //这里我觉得是我代码中最值得称赞的地方
     //从这里可以看到刷题的影子 
     ncpy(ch,&str[j],++index,fp2,&countc);
     for(int i=1; i<=n; i++)
      if(!strcmp(ch,HC[i]))
      {      
       fprintf(fp2,"%c",'a'+i-1);
       j+=index;
       index=2;
       //判别一个字符成功，继续判别下一个字符 
       //改变j的值，使从下一段01代码中复制
       //重新赋值index=2； 
       countc=1;
      }
    }
    fclose(fp1);
    fclose(fp2); 
    printf("\n\t\t译码成功！\n");
    printf("\n\t\t译码结果保存在result.txt文件中！\n");
    printf("\n\t\t请查看！ ");
    system("pause");
    menu(); 
} 

void ncpy(char ch[],char* str,int index,FILE* fp,int* countc)
//自己写的复制字符串函数，舍掉了字符串中的符号 
{
    int k=0;
    for(int i=0; i<index; i++)
     if(str[i]=='1'||str[i]=='0')
      ch[k++]=str[i];
     else if(*countc)
     {
      fprintf(fp,"%c",str[i]);
      *countc=0;
      //这里是我完成程序前的最后一关
      //用countc控制即使多次调用函数，也只输出一次符号 
     }
    ch[k]='\0';
}

继续加油~！~！