数据结构实验二：哈夫曼编码算法的实现

一、实验目的：

1．掌握二叉树的定义；
2．掌握哈夫曼树和哈夫曼编码算法的实现。

二、实验内容：

实现一个哈夫曼编码系统，系统包括以下功能：
(1)给定字符a,b,c,d,e,f的权重为6,2,7,10,3,12。
(2) 建立哈夫曼树：根据统计结果建立哈夫曼树。
(3) 建立哈夫曼码编码：利用得到的哈夫曼树，将各字符对应的编码输出屏幕。

#include <iostream>
using namespace std;

typedef struct {
    int weight;
    int parent,lchild,rchild;
}HTNode,*HuffmanTree;

void select(HuffmanTree HT,int len,int &s1,int &s2){
    int i,min1=0x3f3f3f,min2=0x3f3f3f;  //0x3f3f3f表示无穷大
    for(i=1;i<=len;i++)
    {
        if(HT[i].parent==0&&HT[i].weight<min1)
        {
            min1=HT[i].weight;
            s1=i;
        }
    }
    int temp=HT[s1].weight;
    HT[s1].weight=min2;
    for(i=1;i<=len;i++)
    {
        if(HT[i].parent==0&&HT[i].weight<min2)
        {
            min2=HT[i].weight;
            s2=i;
        }
    }
    HT[s1].weight=temp;
}

构建哈夫曼树

//n表示叶子结点的个数
void CreatHuffmanTree(HuffmanTree &HT,int n){
    if(n<=1) return;
    int m=2*n-1;
    HT=new HTNode[m+1];  //0不放任何结点,HT[m]放根节点
    for(int i=1;i<=m;i++)
    {
        HT[i].lchild=0;
        HT[i].rchild=0;
        HT[i].parent=0;
    }
     cout<<"请输入叶子结点的权值："<<endl;
    for(int i=1;i<=n;++i)
        cin>>HT[i].weight;
    int s1,s2;
    for(int i=n+1;i<=m;++i)
    {
        select(HT,i-1,s1,s2);
        HT[i].lchild=s1;
        HT[i].rchild=s2;
        HT[i].weight=HT[s1].weight+HT[s2].weight;
        HT[s1].parent=i;
        HT[s2].parent=i;
    }
}

返回哈夫曼编码

typedef char **HuffmanCode;     
void CreatHuffmanCode(HuffmanTree HT,HuffmanCode &HC,int n)//存储哈夫曼编码
{
    int i,start,c,f;
    HC=new char*[n+1];  //分配n个字符的头指针矢量，HC为一维的字符头指针数组
    char *cd=new char[n];   //分配临时存放编码的动态数组空间，cd为一维的字符数组的头指针
    cd[n-1]='\0';   //编码结束符
    for(i=1;i<=n;++i)
    {
        start=n-1;
        c=i;
        f=HT[i].parent;
        while(f!=0)
        {           //从叶节点向根节点回溯
            --start;        //回溯一次start向前移一个
            if(HT[f].lchild==c)     //结点c是f的左孩子添0
                cd[start]='0';
            else
                cd[start]='1';      //结点c是f的右孩子添1
            c=f;                    //继续找双亲结点
            f=HT[f].parent;
        }
        HC[i]=new char[n-start];
        strcpy(HC[i], &cd[start]);
    }
    delete[] cd;
}

void show(HuffmanTree HT,HuffmanCode HC)    //输出哈夫曼编码
{
    for(int i=1;i<=sizeof(HC)-2;i++)
        cout<<HT[i].weight<<"编码为"<<HC[i]<<endl;
}

主函数

int main(){
    HuffmanTree HT;
    HuffmanCode HC;
    int n;
    cout<<"请输入叶子结点的个数："<<endl;
    cin>>n;
    CreatHuffmanTree(HT,n);
    CreatHuffmanCode(HT,HC,n);
    cout<<"建立完毕"<<endl;
    show(HT,HC);
    return 0;
}

案例结果展示：