haffmantree.c

程序通过构建一个haffman树

        }
        if(!ptr->lchild&&!ptr->rchild)     //如果没有左右孩子，即叶子结点
        {
          ptr->mark=1;
          code[index++]='\0';
          codedictionary[k].ch=ptr->ch;   //给字典赋字符值
          for(index=0;code[index]!='\0';index++)
            codedictionary[k].code[index]=code[index];//给字典赋码值
          codedictionary[k].code[index]='\0';
          k++;
          ptr=tree;
          index=0;
        }
        if(ptr->lchild->mark==1&&ptr->rchild->mark==1)//如果左右孩子都已标记
        {
          ptr->mark=1;
          ptr=tree;
          index=0;
        }
    }
  }
  printf("\n");

}

/**
*解码，即将0-1码转化为字符串
*输入：（哈夫曼结点指针）编码树的地址
*     （字符型指针）要解码的0-1字符串地址
*     （字符型指针）解码后的字符串地址
*返回：空
*/


void decodeHTree(LinkTree tree,char *code,char *filestring)
{
    int i=0,j=0,k=0;
    char *char0_1;
    LinkTree ptr=tree;
    char0_1=(char *)malloc(MAX_SINGLECODE_LEN); //此数组用于统计输入的0-1序列

    printf("预览解压后的字符：\n");
 //printf("%s\n",code);
    for(j=0,ptr=tree;code[i]!='\0'&&ptr->lchild&&ptr->rchild;j=0,ptr=tree)
 {
  for(j=0;code[i]!='\0'&&ptr->lchild&&ptr->rchild;j++,i++)
  {
            if(code[i]=='0')
   {
                ptr=ptr->lchild;
                char0_1[j]='0';
            }
            if(code[i]=='1')
   {
                ptr=ptr->rchild;
                char0_1[j]='1';
            }
        }
        if(!ptr->lchild&&!ptr->rchild)
  {
   printf("%c",ptr->ch);    //显示解压后的字符
   filestring[k++]=ptr->ch;   //将字符逐一保存到字符串里
  }
  if(code[i]=='\0'&&ptr->lchild&&ptr->rchild)
  {
            char0_1[j]='\0';
            printf("\n没有与最后的几个0-1序列：%s相匹配的字符!\n",char0_1);
            return;
        }
    }
 printf("\n\n");
 filestring[k]='\0';
    free(char0_1);
 printf ("st\t");
}

 

/**
*释放哈夫曼树所占用的空间
*输入：（哈夫曼结点指针）要释放的结点地址
*返回：空
*/
void deleteNode(LinkTree tree)
{
  LinkTree ptr=tree;
  if(ptr)
  {
    deleteNode(ptr->lchild);
    deleteNode(ptr->rchild);
    free(ptr);
  }
}

/**
*将整个字符串转化为0-1的字符串
*输入：（字符型指针）待转化的字符串首地址
*     （编码字典类型指针）字典首地址
*     （字符型指针）接收0-1码串的首地址
*返回：空
*/
void compressString(char *string,CodeDictionary *codedictionary,char *codestring)
{
 int i=0,j=0,k=0,m;

 while(string[i])       //整个文件字符串没结束时
 {
  while(string[i]!=codedictionary[j].ch&&j<MAX_WORDS)
            //找与对应字符相同的字符
   j++;
  if(string[i]==codedictionary[j].ch)  //如果找到与对应字符相同的字符
   for(m=0;codedictionary[j].code[m];m++,k++)
    codestring[k]=codedictionary[j].code[m];
  j=0;         //字典复位
  i++;
 }
 codestring[k]='\0';
}


/**
*把指定文件读到字符串中
*输入：（字符型指针）待接收文件的字符串地址
*返回：空
*/
void readFile(char *string)
{
  FILE *fp;
  int i;
  char ch;                         //记录读入的字符
  char path[PATH_LEN];                 //文本文件的读路径

  printf("请输入要压缩的文本文件地址：（无需扩展名）");
  gets(path);
  if((fp=fopen(strcat(path,".txt"),"r"))==NULL)
  {
    printf("\n路径不正确！\n");
    getch();
    return;
  }

  ch=fgetc(fp);
  for(i=0;ch!=EOF;i++)
  {
    string[i]=ch;
    ch=fgetc(fp);
  }
  string[i+1]='\0';
  //printf("%s",string);
  fclose(fp);
}

/**
*保存编码后的解码树和字符串
*输入：（哈夫曼结点指针）解码树的地址
*     （字符型指针）要保存的0-1码串首地址
*返回：空
*/
void writeCode(LinkTree tree,char *string)
{
 FILE *fp;
 int i;
 int weight;         //记录写入的权值
 char ch;         //记录写入的字符
 LinkTree p;
 char path[PATH_LEN];      //0-1码文件的写路径

 printf("请输入压缩后的保存路径及文件名：（无需扩展名）");
 gets(path);
 if((fp=fopen(strcat(path,".txt"),"w+"))==NULL)
 {
  printf("\n文件路径出错！\n");
  getch();
  return;
 }
 //printf("%s\n",string);
 p=tree->next;

 /*解码树部分写入文件前部分*/
 do
 {
  ch=p->ch;
  weight=p->weight;
  fprintf(fp,"%c%d  ",ch,weight);
  p=p->next;
 }while(p);
 fprintf(fp,"%c%d",'^',END_TREE);

 fseek(fp,sizeof(char),1);     //空出区分位

 /*0-1码写入文件后部分*/
 for(i=0;string[i];i++)
 {
  ch=string[i];
  fputc(ch,fp);
 }
 printf("\n压缩成功！\n");
 getch();
 fclose(fp);
}

/**
*读取编码后的0-1字符串
*输入：（哈夫曼结点指针）解码树的地址
*     （字符型指针）要接收的0-1码串首地址
*返回：空
*/
void readCode(LinkTree tree,char *string)
{

  FILE *fp;
  int i;
  int weight;                         //记录读入的权值
  char ch,blank;                         //记录读入的字符
  LinkTree ptr,beforeptr;
  char path[PATH_LEN];                 //0-1码文件的读路径

  printf("请输入要解压的文件路径及文件名：（无需扩展名）");
  gets(path);
  if((fp=fopen(strcat(path,".txt"),"r"))==NULL)
  {
    printf("\n文件路径出错！\n");
    getch();
    return;
  }
  beforeptr=tree;

  /*从文件前部分读出解码树*/
  fscanf(fp,"%c%d",&ch,&weight);
  while(weight!=END_TREE)
  {
    if((ptr=(LinkTree)malloc(sizeof(HTNode)))==NULL)
    {
        printf("内存不足！");
        getch();
        exit(1);                   //错误出口
    }
    ptr->ch=ch;
    ptr->weight=weight;
    ptr->lchild=NULL;
    ptr->rchild=NULL;
    ptr->parent=NULL;
    ptr->mark=0;
    beforeptr->next=ptr;
    beforeptr=ptr;
 fscanf(fp,"%c%c",&blank,&blank);//除去多余的空格
    fscanf(fp,"%c%d",&ch,&weight);
  }
  beforeptr->next=NULL;

  fseek(fp,sizeof(char),1);             //文件指针定位

  /*从文件后部分读出0-1码*/

  ch=fgetc(fp);
  for(i=0;ch!=EOF;i++)
  {
 
    string[i]=ch;
    ch=fgetc(fp);
  }
  string[i]='\0';
  fclose(fp);
 }

/**
*保存解码后的文件
*输入：（字符型指针）解码后的字符串地址
*返回：空
*/
void writeFile(char *string)
{
  FILE *fp;
  char ch;                         //记录写入的字符
  int i;
  char path[PATH_LEN];                 //文本文件的写路径
  //printf ("%s",string);

  printf("请输入解压后的保存路径及文件名：（无需扩展名）");
  gets(path);
  if(!path[0] || path[0]==' ') strcpy(path,"default");
  if((fp=fopen(strcat(path,".txt"),"w+"))==NULL)
  {
    printf("\n文件路径出错！\n");
    getch();
    return;
  }

  printf("start\n");
  for(i=0;string[i];i++)
  {
    ch=string[i];
    fputc(ch,fp);
  }
  printf("\n解压成功！\n");
  getch();
  fclose(fp);
}

/**
*显示主菜单
*输入：空
*返回：空
*/
void menu()
{
  printf("\n\n\n\n\n\n");
  printf("\t\t -----**   欢迎使用WINYXY压缩工具   **-----");
  printf("\n\n\n");
  printf("\t\t\t\t<c> 压 缩\n\n");
  printf("\t\t\t\t<u> 解 压\n\n");
  printf("\t\t\t\t<e> 退 出\n\n");
  printf("\n\n请按键选择：\n");
  printf("\n\n\n\n\n\n");
}