1. 程式人生 > >[C語言]哈夫曼樹(Huffman)的構造與實現

[C語言]哈夫曼樹(Huffman)的構造與實現

     C語言資料結構中哈夫曼樹是個重要的內容。哈夫曼主要是它的編碼應用可以保證譯碼的非二義性。

    每天堅持編寫一個程式,持之以恆,我們就會更加熟練的進行程式設計,從而為以後打下基礎。

    下面是今天編寫的HUffman樹的原始碼,因為純手寫,沒有執行,瞭解了原理才是最重要的嘛!

#include <stdio.h>
#include <stdlib.h>
#include <string.h>


#define n 6
#define m 2*n-1

typedef struct 
{
float weight;
int lchild,rchild,parent;
}codenode;

typedef codenode huffmantree[m];

typedef struct
{
char ch;
char bits[n+1];
}code;

typedef code huffmancode[n];

//哈夫曼樹的初始化
void inithf(huffmantree T)
{
int i;
for (i=0;i<m;i++)
{
T[i].weight=0;
T[i].parent=-1;
T[i].lchild=-1;
T[i].rchild=-1;
}
}


//輸入哈夫曼的資料
void inputhf(huffmantree T)
{
int i;
float k;
for (i=0;i<n;i++)
{
scanf("%f",&k);
T[i].weight=k;
}
}


void selectmin(huffmantree T,int k,int *p1,int *p2)
{
int i;
float small1=10000,small2=10000;
for(i=0;i<k;i++)
{
if (T[i].parent==-1)
if(T[i].weight<small1)
{
small2=small1;
small1=T[i].weight;
*p2=*p2;
*p1=i;
}
else
if (T[i].weight<small2)
{
small2=T[i].weight;
*p2=i;
}
}
}


//建立哈夫曼樹
void creathftree(huffmantree T)
{
int i,p1,p2;
inithf(T);
inputhf(T);
for(i=n;i<m;i++)
{
selectmin(T,i-1,&p1,&p2);
T[p1].parent=T[p2].parent=i;
T[i].lchild=p1;
T[i].rchild=p2;
T[i].weight=T[p1].weight+T[p2].weight;
}

}


//哈夫曼編碼表
void creathfcode(huffmantree T,huffmancode H)
{
int c,p,i,start;
char cd[n+1];
cd[n]='\0';
for(i=0;i<n;i++)
{
H[i].ch=getchar();
start=n;
c=i;
while((p=T[c].parent)>=0)
{
cd[--start]=(T[p].lchild==c)?'0':'1';
c=p;
}
strcpy(H[i].bits,&cd[start]);
}
}


//編碼
void zip(huffmancode H,char *ch,char *s)
{
int j=0;
char *p[n];
unsigned int i;
for (i=0;i<strlen(ch);i++)
{
while (ch[i]!=H[j].ch) j++;
p[i]=H[j].bits;
}
strcpy(s,p[0]);
for (i=1;i<n;i++)
strcat(s,p[i]);
puts(s);
}


//解碼
void uzip(huffmancode H,char *s,huffmantree T)
{
int j=0,p;
unsigned int i;
char ch[n+1];
while (i<strlen(s))
{
p=m-1;
while (T[p].lchild!=-1)
{
if(s[i]=='0')
{
p=T[p].lchild;
i++;
}
else
{
p=T[p].rchild;
i++;
}
}
ch[j]=H[p].ch;
j++;
}
ch[j]='\0';
puts(ch);
}


void main()
{
char ch[]="abcdef",s[36];
huffmantree T;
huffmancode H;
creathftree(T);
getchar();
creathfcode(T,H);
zip(H,ch,s);
uzip(H,s,T);
}
希望大家可以一起多交流,一起進步哦。

文章在我論壇的地址:http://www.jsjer.com/forum.php?mod=viewthread&tid=762