用C++实现哈夫曼编码

@用C++实现哈夫曼编码
哈夫曼编码

霍夫曼编码使用变长编码表对源符号（如文件中的一个字母）进行编码，其中变长编码表是通过一种评估来源符号出现机率的方法得到的，出现机率高的字母使用较短的编码，反之出现机率低的则使用较长的编码，这便使编码之后的字符串的平均长度、期望值降低，从而达到无损压缩数据的目的。

问题描述

给定报文中26个字母a-z及空格的出现频率{64, 13, 22, 32, 103, 21, 15, 47, 57, 1, 5, 32, 20, 57, 63, 15, 1, 48, 51, 80, 23, 8, 18, 1, 16, 1, 168}，构建哈夫曼树并为这27个字符编制哈夫曼编码，并输出。模拟发送端，从键盘输入字符串，以%为结束标记，在屏幕上输出输入串的编码；模拟接收端，从键盘上输入0-1哈夫曼编码串，翻译出对应的原文。

代码实现

// An highlighted block
var foo = 'bar';
#include<bits/stdc++.h>
using namespace std;
typedef struct HTNode
{
    int weight;
    int parent,lchild,rchild;
} HTNode, *HuffmanTree;

typedef char **HuffmanCode;
int i,j,m,n,s1,s2,w[100];
void select(HuffmanTree HT,int n, int &s1, int &s2)
{
    for(i=1; i<=n; i++)
    {
        if(HT[i].parent==0)
        {
            s1=i;
            break;
        }
    }
    for(i=1; i<=n; i++)
    {
        if(HT[i].parent==0 && HT[s1].weight>HT[i].weight)
        {
            s1=i;
        }
    }
    for(i=1; i<=n; i++)
    {
        if(HT[i].parent==0 && i!=s1)
        {
            s2=i;
            break;
        }
    }
    for(i=1; i<=n; i++)
    {
        if(HT[i].parent==0 && HT[s2].weight>HT[i].weight && i!=s1)
        {
            s2=i;
        }
    }
}

void createhuffmantree(HuffmanTree &HT,int n)
{
    if(n<=1)
    {
        return;
    }
    m=(2*n)-1;
    HT=new HTNode[m+1];
    for(i=1; i<=m; i++)
    {
        HT[i].parent=0;
        HT[i].lchild=0;
        HT[i].rchild=0;
    }
    for(i=1; i<=n; i++)
    {
        cin>>HT[i].weight;
    }
    for(i=n+1; i<=m; i++)
    {
        select(HT,i-1,s1,s2);
        HT[s1].parent=i;
        HT[s2].parent=i;
        HT[i].lchild=s1;
        HT[i].rchild=s2;
        HT[i].weight=HT[s1].weight+HT[s2].weight;
    }
}
void creathuffmancode(HuffmanTree HT,HuffmanCode &HC,int n)
{
    int start,c;
    int f;
    HC=new char*[n+1];
    char *cd=new char[n];
    cd[n-1]='\0';
    for(i=1; i<=n; i++)
    {
        start=n-1;
        c=i;
        f=HT[c].parent;
        while(f!=0)
        {
            start--;
            if(HT[f].lchild==c)
            {
                cd[start]='0';
            }
            else
            {
                cd[start]='1';
            }
            c=f;
            f=HT[f].parent;
        }
        HC[i]=new char[n-start];
        strcpy(HC[i],&cd[start]);
    }
    delete cd;
}

void TranCode(HuffmanTree HT,char a[],char zf[],char b[],int n)
{
    int q=2*n-1;
    int k=0;
    for(i=0; a[i]!='\0'; i++)
    {
        if(a[i]=='0')
        {
            q=HT[q].lchild;
        }
        else if(a[i]=='1')
        {
            q=HT[q].rchild;
        }
        if(HT[q].lchild==0 && HT[1].rchild==0)
        {
            b[k++]=zf[q];
            q=2*n-1;
        }
    }
    b[k]='\0';
}
void show_help()
{
    cout<<"********************************************************************"<<endl;
    cout<<"********* 1.输入HuffmanTree的参数.                        **********"<<endl;
    cout<<"********* 2.初始化HuffmanTree参数(含有26字母及空格).      **********"<<endl;
    cout<<"********* 3.创建HuffmanTree和编码表.                      **********"<<endl;
    cout<<"********* 4.输出编码表.                                   **********"<<endl;
    cout<<"********* 5.输入编码，并翻译为字符.                       **********"<<endl;
    cout<<"********* 6.输入字符，并实现编码.                         **********"<<endl;
    cout<<"********* 7.退出.                                         **********"<<endl;
    cout<<"********************************************************************"<<endl;
}

int main()
{
    HuffmanTree HT;
    HuffmanCode HC;
    int operator_code;
    char a[100];
    char b[100];
    char c[100];
    char zf[100];
    show_help();
    while(1)
    {
        cout<<"请输入操作代码："<<endl;
        cin>>operator_code;
        if(operator_code==1)
        {
            cout<<"请输入需要编码的字符个数："<<endl;
            cin>>n;
        }
        else if(operator_code==2)
        {
            cout<<"请输入所有编码的字符："<<endl;
            for(i=1; i<=n; i++)
            {
                cin>>zf[i];
            }
            cout<<"请输入每一个字符的权值："<<endl;
            createhuffmantree(HT,n);
            cout<<"结点"<<"\t"<<"字符"<<"\t"<<"权值"<<"\t"<<"双亲"<<"\t"<<"左孩子"<<"\t"<<"右孩子"<<endl;
            for(int i=1;i<=n;i++)
            {
                cout<<i<<"\t"<<zf[i]<<"\t"<<HT[i].weight<<"\t"<<HT[i].parent<<"\t"<<HT[i].lchild<<"\t"<<HT[i].rchild<<endl;
            }
        }
        else if(operator_code==3)
        {
            creathuffmancode(HT,HC,n);
        }
        else if(operator_code==4)
        {
            cout<<"结点"<<"\t"<<"字符"<<"\t"<<"权值"<<"\t"<<"编码"<<endl;
            for(int i=1;i<=n;i++)
            {
                cout<<i<<"\t"<<zf[i]<<"\t"<< HT[i].weight<<"\t"<<HC[i]<<endl;
            }
            cout<<endl;
        }
        else if(operator_code==5)
        {
            cout<<"输入二进制编码："<<endl;
            cin>>a;
            TranCode(HT,a,zf,b,n);
            cout<<"结果为："<<b<<endl;

        }else if(operator_code==6)
        {
            cout<<"请输入一串字符"<<endl;
            cin>>c;
            for(i=0;i<sizeof(c)/sizeof(c[0]);i++)
            {
                for(j=1;i<=n;j++)
                {
                    if(c[i]==zf[j])
                    {
                        cout<<HC[j];
                        break;
                    }
                }
            }
            cout<<endl;
        }else if(operator_code==7)
        {
            break;
        }
    }
    return 0;
}

实现效果