huffman数据结构书上的

#include<limits.h> /* INT_MAX等 */
#include<stdio.h> /* EOF(=^Z或F6),NULL */
#include <malloc.h>
#include <string.h>
#define MAX 100
typedef struct{
unsigned int weight;
unsigned int parent,lchild,rchild;
}HTNode,*HuffmanTree; /* 动态分配数组存储赫夫曼树 */
typedef char **HuffmanCode; /* 动态分配数组存储赫夫曼编码表 */
int min1(HuffmanTree t,int i){
/* 函数void select()调用 */
int j,minpos;
unsigned int min=UINT_MAX; /* 取k为不小于可能的值 */
for (j=1;j<=i;j++)
if(t[j].weight<min&&t[j].parent==0)\
min=t[j].weight,minpos=j;
t[minpos].parent=i;
return minpos;
}
void select(HuffmanTree t,int i,int *s1,int *s2){
/* s1为最小的两个值中序号小的那个 */
int j;
*s1=min1(t,i);
*s2=min1(t,i);
if(*s1>*s2){
j=*s1;
*s1=*s2;
*s2=j;
}
}
void HuffmanCoding(HuffmanTree &HT,HuffmanCode &HC,int num[],int n){
/* w存放n个字符的权值(均>0),构造赫夫曼树HT,并求出n个字符的赫夫曼编码HC */
int m,i,s1,s2,start;
unsigned c,f;
HuffmanTree p;
char *cd;
if(n<=1)
return;
m=2*n-1;
HT=(HuffmanTree)malloc((m+1)*sizeof(HTNode)); /* 0号单元未用 */
for(p=HT+1,i=1;i<=n;++i,++p){
(*p).weight=num[i-1]; (*p).parent=0;
(*p).lchild=0; (*p).rchild=0;
}
for(;i<=m;++i,++p){


(*p).weight=0;
(*p).parent=0;
(*p).lchild=0;
(*p).rchild=0;
}
for(i=n+1;i<=m;++i){ /* 建赫夫曼树 */
/* 在HT[1~i-1]中选择parent为0且weight最小的两个结点,其序号分别为s1和s2 */
select(HT,i-1,&s1,&s2);
HT[s1].parent=HT[s2].parent=i;
HT[i].lchild=s1;
HT[i].rchild=s2;
HT[i].weight=HT[s1].weight+HT[s2].weight;
}
/* 从叶子到根逆向求每个字符的赫夫曼编码 */
HC=(HuffmanCode)malloc((n+1)*sizeof(char*));
/* 分配n个字符编码的头指针向量([0]不用) */
cd=(char*)malloc(n*sizeof(char)); /* 分配求编码的工作空间 */
cd[n-1]='\0'; /* 编码结束符 */
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]=(char*)malloc((n-start)*sizeof(char));
/* 为第i个字符编码分配空间 */
strcpy(HC[i],&cd[start]); /* 从cd复制编码(串)到HC */
}
free(cd); /* 释放工作空间 */
}




void detect(char s[],char ch[],int num[])
{

int i,j,n=0;
for(i=0;s[i]!='\0';i++)
{
for(j=0;j<n;j++)
if(s[i]==ch[j]||(ch[j]>='a'&&ch[j]<'z'&&s[i]+32==ch[j]))break;
//判断该字符是否已经出现过
if(j<n)//该字符出现过，对应的计数器num[j]加一
num[j]++;
   else   //如果该字符是新出先的字符，记录到ch[j]中，对应的计数器加一
   {
   if(s[i]>='A'&&s[i]<='Z')
   ch[j]=s[i]+32;
   else
   ch[j]=s[i];
        num[j]++;
n++;
   }
}
for(i=0;i<n;i++)
printf("\'%c\'对应的密码是：\n",ch[i]);
}










void main()
{
HuffmanTree HT;
HuffmanCode HC;
int n=0,i;
char s[MAX];
char ch[MAX];//记录出现的字符
    int num[MAX]={0};//记录每个字符出现的次数
printf("请输入一个密码加密：\n");
scanf("%s",s);
detect(s,ch,num);
for(i=0;num[i]!='\0';i++)
{
n=n+1;
}
  
HuffmanCoding(HT,HC,num,n);
for(i=1;i<=n;i++)
   puts(HC[i]);

}