哈夫曼樹與哈夫曼編碼

定義:

給定n個權值作爲n個葉子結點,構造一棵二叉樹,若帶權路徑長度達到最小,稱這樣的二叉樹爲最優二叉樹,也稱爲哈夫曼樹(Huffman tree)。哈夫曼樹是帶權路徑長度最短的樹,權值較大的結點離根較近。

構造:

1.構成初始集合
  對給定的n個權值{W1,W2,W3,...,Wi,...,Wn}構成n棵二叉樹的初始集合F={T1,T2,T3,...,Ti,...,Tn},其中每棵二叉樹Ti中只有一個權值爲Wi的根結點,它的左右子樹均爲空。(爲方便在計算機上實現算法,一般還要求以Ti的權值Wi的升序排列。) 
2.選取左右子樹
  在F中選取兩棵根結點權值最小的樹作爲新構造的二叉樹的左右子樹,新二叉樹的根結點的權值爲其左右子樹的根結點的權值之和。 
3.刪除左右子樹
  從F中刪除這兩棵樹,並把這棵新的二叉樹同樣以升序排列加入到集合F中。 
4.重複二和三兩步,直到集合F中只有一棵二叉樹爲止。

c++代碼實現

#include<iostream>
#include<stdlib.h>
using namespace std;
const int MaxValue =10000;//初始設定的權值最大值
const int MaxBit =4;//初始設定的最大編碼位數
const int MaxN=10;//初始設定的最大結點個數
struct HaffNode//哈夫曼樹的結點結構
{
	int weight;//權值
	int flag;//標記
	int parent;//雙親結點下標
    int leftChild;//左孩子下標
    int rightChild;//右孩子下標
};
struct Code//存放哈夫曼編碼的數據元素結構
{
	int bit[MaxBit];//數組
	int start;//編碼的起始下標
	int weight;//字符的權值
};
//weight[]:由小到大排序
void Haffman(int weight[], int n, HaffNode haffTree[])
//建立葉結點個數爲n權值爲weight的哈夫曼樹haffTree
{
	int j,m1,m2,x1,x2;
   //哈夫曼樹haffTree初始化。n個葉結點的哈夫曼樹共有2n-1個結點
	for(int i=0;i<2*n-1;i++)
	{
		if(i<n)
			haffTree[i].weight=weight[i];
		else
			haffTree[i].weight=0;
        //注意這裏沒打else那{},故無論是n個葉子節點還是n-1個非葉子節點都會進行下面4步的初始化
		haffTree[i].parent=0;
		haffTree[i].flag=0;
		haffTree[i].leftChild=-1;
		haffTree[i].rightChild=-1;
	}
	//構造哈夫曼樹haffTree的n-1個非葉結點
	for(int i=0;i<n-1;i++)
	{
		m1=m2=MaxValue;//Maxvalue=10000;(就是一個相當大的數)
		x1=x2=0;//x1、x2是用來保存最小的兩個值在數組對應的下標
		for(j=i;j<n+i;j++)//循環找出所有權重中,最小的二個值--morgan
		{
			if(haffTree[j].weight<m1&&haffTree[j].flag==0)
			{
				m2=m1;
				x2=x1;
				m1=haffTree[j].weight;
				x1=j;
			}
			else if(haffTree[j].weight<m2&&haffTree[j].flag==0)
			{
				m2=haffTree[j].weight;
				x2=j;
			}
		}
		cout<<"i="<<i<<" "<<m1<<" "<<m2<<endl;
		//將找出的兩棵權值最小的子樹合併爲一棵子樹
		haffTree[x1].parent=n+i;
		haffTree[x2].parent=n+i;
		haffTree[x1].flag=1;
		haffTree[x2].flag=1;
		haffTree[n+i].weight=haffTree[x1].weight+haffTree[x2].weight;
		haffTree[n+i].leftChild=x1;
		haffTree[n+i].rightChild=x2;
	}
}
void HaffmanCode(HaffNode haffTree[],int n,Code haffCode[])
//由n個結點的哈夫曼樹haffTree構造哈夫曼編碼haffCode
{
	Code*cd=new Code;
	int child, parent;

	//求n個葉結點的哈夫曼編碼
	for(int i=0;i<n;i++)
	{
		//cd->start=n-1;//不等長編碼的最後一位爲n-1,
		cd->start=0;//,----修改從0開始計數--morgan
		cd->weight=haffTree[i].weight;//取得編碼對應權值的字符
		child=i;
		parent=haffTree[child].parent;
		//由葉結點向上直到根結點
		while(parent!=0)
		{
			if(haffTree[parent].leftChild==child)
				cd->bit[cd->start]=0;//左孩子結點編碼0
			else
				cd->bit[cd->start]=1;//右孩子結點編碼1
			//cd->start--;
			cd->start++;//改成編碼自增--morgan
			child=parent;
			parent=haffTree[child].parent;
		}
		//保存葉結點的編碼和不等長編碼的起始位
		//for(intj=cd->start+1;j<n;j++)
		for(int j=cd->start-1;j>=0;j--)//重新修改編碼,從根節點開始計數--morgan
			haffCode[i].bit[cd->start-j-1]=cd->bit[j];
		
		haffCode[i].start=cd->start;
		haffCode[i].weight=cd->weight;//保存編碼對應的權值
	}
}
int main()
{
	int i, j, n=4,m=0;
    int weight[]={2,4,5,7};
	//int weight[]={7,5,4,2};
    HaffNode*myHaffTree=new HaffNode[2*n-1];
    Code*myHaffCode=new Code[n];
    if(n>MaxN)
	{
		cout<<"定義的n越界,修改MaxN!"<<endl;
		exit(0);
	}
	Haffman(weight,n,myHaffTree);
	HaffmanCode(myHaffTree,n,myHaffCode);
    //輸出每個葉結點的哈夫曼編碼
	for(i=0;i<n;i++)
	{
		cout<<"Weight="<<myHaffCode[i].weight<<" Code=";
		//for(j=myHaffCode[i].start+1;j<n;j++)
		for(j=0;j<myHaffCode[i].start;j++)
			cout<<myHaffCode[i].bit[j];
		m=m+myHaffCode[i].weight*myHaffCode[i].start;
		cout<<"   長度:"<<myHaffCode[i].start<<endl;
	}
	cout<<"huffman'sWPLis:";
	cout<<m;
	cout<<endl;
	return 0;
}

c代碼實現:

/*-------------------------------------------------------------------------
 * Name:   哈夫曼編碼源代碼。
 * Date:   2011.04.16
 * Author: Jeffrey Hill+Jezze(解碼部分)
 * 在 Win-TC 下測試通過
 * 實現過程:着先通過 HuffmanTree() 函數構造哈夫曼樹,然後在主函數 main()中
 *           自底向上開始(也就是從數組序號爲零的結點開始)向上層層判斷,若在
 *           父結點左側,則置碼爲 0,若在右側,則置碼爲 1。最後輸出生成的編碼。
 *------------------------------------------------------------------------*/
#include <stdio.h>
#include<stdlib.h>
 
#define MAXBIT      100
#define MAXVALUE  10000
#define MAXLEAF     30
#define MAXNODE    MAXLEAF*2 -1
 
typedef struct 
{
    int bit[MAXBIT];
    int start;
} HCodeType;        /* 編碼結構體 */
typedef struct
{
    int weight;
    int parent;
    int lchild;
    int rchild;
    int value;
} HNodeType;        /* 結點結構體 */
 
/* 構造一顆哈夫曼樹 */
void HuffmanTree (HNodeType HuffNode[MAXNODE],  int n)
{ 
    /* i、j: 循環變量,m1、m2:構造哈夫曼樹不同過程中兩個最小權值結點的權值,
        x1、x2:構造哈夫曼樹不同過程中兩個最小權值結點在數組中的序號。*/
    int i, j, m1, m2, x1, x2;
    /* 初始化存放哈夫曼樹數組 HuffNode[] 中的結點 */
    for (i=0; i<2*n-1; i++)
    {
        HuffNode[i].weight = 0;//權值 
        HuffNode[i].parent =-1;
        HuffNode[i].lchild =-1;
        HuffNode[i].rchild =-1;
        HuffNode[i].value=i; //實際值,可根據情況替換爲字母  
    } /* end for */
 
    /* 輸入 n 個葉子結點的權值 */
    for (i=0; i<n; i++)
    {
        printf ("Please input weight of leaf node %d: \n", i);
        scanf ("%d", &HuffNode[i].weight);
    } /* end for */
 
    /* 循環構造 Huffman 樹 */
    for (i=0; i<n-1; i++)
    {
        m1=m2=MAXVALUE;     /* m1、m2中存放兩個無父結點且結點權值最小的兩個結點 */
        x1=x2=0;
        /* 找出所有結點中權值最小、無父結點的兩個結點,併合並之爲一顆二叉樹 */
        for (j=0; j<n+i; j++)
        {
            if (HuffNode[j].weight < m1 && HuffNode[j].parent==-1)
            {
                m2=m1; 
                x2=x1; 
                m1=HuffNode[j].weight;
                x1=j;
            }
            else if (HuffNode[j].weight < m2 && HuffNode[j].parent==-1)
            {
                m2=HuffNode[j].weight;
                x2=j;
            }
        } /* end for */
            /* 設置找到的兩個子結點 x1、x2 的父結點信息 */
        HuffNode[x1].parent  = n+i;
        HuffNode[x2].parent  = n+i;
        HuffNode[n+i].weight = HuffNode[x1].weight + HuffNode[x2].weight;
        HuffNode[n+i].lchild = x1;
        HuffNode[n+i].rchild = x2;
 
        printf ("x1.weight and x2.weight in round %d: %d, %d\n", i+1, HuffNode[x1].weight, HuffNode[x2].weight);  /* 用於測試 */
        printf ("\n");
    } /* end for */
  /*  for(i=0;i<n+2;i++)
    {
        printf(" Parents:%d,lchild:%d,rchild:%d,value:%d,weight:%d\n",HuffNode[i].parent,HuffNode[i].lchild,HuffNode[i].rchild,HuffNode[i].value,HuffNode[i].weight);
                  }*///測試 
} /* end HuffmanTree */
 
//解碼 
void decodeing(char string[],HNodeType Buf[],int Num)
{
  int i,tmp=0,code[1024];
  int m=2*Num-1;
  char *nump;
  char num[1024];
  for(i=0;i<strlen(string);i++)
  {
   if(string[i]=='0')
  num[i]=0;        
  else
  num[i]=1;                    
  } 
  i=0;
  nump=&num[0];
  
 while(nump<(&num[strlen(string)]))
 {tmp=m-1;
  while((Buf[tmp].lchild!=-1)&&(Buf[tmp].rchild!=-1))
  {
  
   if(*nump==0)
   {
     tmp=Buf[tmp].lchild ;          
   } 
   else tmp=Buf[tmp].rchild;
   nump++;
        
  } 
  
  printf("%d",Buf[tmp].value);                                  
 }
 
  
}
 
 
int main(void)
{
    
    HNodeType HuffNode[MAXNODE];            /* 定義一個結點結構體數組 */
    HCodeType HuffCode[MAXLEAF],  cd;       /* 定義一個編碼結構體數組, 同時定義一個臨時變量來存放求解編碼時的信息 */
    int i, j, c, p, n;
    char pp[100];
    printf ("Please input n:\n");
    scanf ("%d", &n);
    HuffmanTree (HuffNode, n);
   
    
    for (i=0; i < n; i++)
    {
        cd.start = n-1;
        c = i;
        p = HuffNode[c].parent;
        while (p != -1)   /* 父結點存在 */
        {
            if (HuffNode[p].lchild == c)
                cd.bit[cd.start] = 0;
            else
                cd.bit[cd.start] = 1;
            cd.start--;        /* 求編碼的低一位 */
            c=p;                    
            p=HuffNode[c].parent;    /* 設置下一循環條件 */
        } /* end while */
        
        /* 保存求出的每個葉結點的哈夫曼編碼和編碼的起始位 */
        for (j=cd.start+1; j<n; j++)
        { HuffCode[i].bit[j] = cd.bit[j];}
        HuffCode[i].start = cd.start;
    } /* end for */
    
    /* 輸出已保存好的所有存在編碼的哈夫曼編碼 */
    for (i=0; i<n; i++)
    {
        printf ("%d 's Huffman code is: ", i);
        for (j=HuffCode[i].start+1; j < n; j++)
        {
            printf ("%d", HuffCode[i].bit[j]);
        }
        printf(" start:%d",HuffCode[i].start);
       
        printf ("\n");
        
    }
/*    for(i=0;i<n;i++){
    for(j=0;j<n;j++)
        {
             printf ("%d", HuffCode[i].bit[j]);           
        }
        printf("\n");
        }*/
    printf("Decoding?Please Enter code:\n");
    scanf("%s",&pp);
decodeing(pp,HuffNode,n);
    getch();
    return 0;
}

參考:




 



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章