7-2 哈夫曼编码译码 (25 分)

7-2 哈夫曼编码译码 (25 分)

编写一个哈夫曼编码译码程序。

按词频从小到大的顺序给出各个字符（不超过30个）的词频，根据词频构造哈夫曼树，给出每个字符的哈夫曼编码，并对给出的语句进行译码。

为确保构建的哈夫曼树唯一，本题做如下限定：

（1）选择根结点权值最小的两棵二叉树时，选取权值较小者作为左子树。

（2）若多棵二叉树根结点权值相等，按先后次序分左右，先出现的作为左子树，后出现的作为右子树。

生成哈夫曼编码时，哈夫曼树左分支标记为0，右分支标记为1。

输入格式:
第一行输入字符个数n；

第二行到第n行输入相应的字符及其词频(可以是整数，与可以是小数）；

最后一行输入需进行译码的串。

输出格式:
首先按树的先序顺序输出所有字符的编码，每个编码占一行；

最后一行输出需译码的原文，加上original:字样。

输出中均无空格

输入样例:
3
m1
n1
c2
10110
输出样例:
c:0
m:10
n:11
original:mnc

题解如下

#include <iostream>
#include <algorithm>
#include <stdio.h>
using namespace std;
typedef struct TNode *Position;
typedef Position BinTree;
struct TNode{
    char Data;
    double weight;
    BinTree Left;
    BinTree Right;
};
BinTree cNode(char c,double weight){
    BinTree T;
    T = (BinTree)malloc(sizeof(struct TNode));
    T->Left = NULL;
    T->Right = NULL;
    T->Data = c;
    T->weight = weight;
    return T;
}
BinTree cTree(BinTree L,BinTree R){
    BinTree T;
    T = (BinTree)malloc(sizeof(struct TNode));
    T->Left = L;
    T->Right = R;
    T->Data = '-';
    T->weight = L->weight + R->weight;
    return T;
}
bool cmp(BinTree a,BinTree b){
    return a->weight<b->weight;
}
void Pre(BinTree BT, string a){
    if(BT){
        if(BT->Data!='-'){
            printf("%c:",BT->Data);
            cout<<a<<endl;
        }
        Pre(BT->Left, a.append("0"));
        a = a.substr(0,a.length()-1);
        Pre(BT->Right, a.append("1"));
    }
}
void decode(BinTree BT,string a){
    BinTree BT2 = BT;
    for(int i=0;i<a.length();i++){
        if(a[i]=='0'){
            BT2 = BT2->Left;
        }
        if(a[i]=='1'){
            BT2 = BT2->Right;
        }
        if(BT2->Data!='-'&&BT2){
            cout<<BT2->Data;
            BT2 = BT;
        }
    }
}
int main(){
    int n;
    scanf("%d",&n);
    getchar();
    char data[n];
    double weight[n];
    for(int i=0;i<n;i++){
        scanf("%c%lf",&data[i],&weight[i]);
        getchar();
    }
    string dc;
    cin>>dc;
    BinTree T;
    BinTree bt[2*n-1];
    for(int i=0;i<n;i++){
        bt[i] = cNode(data[i],weight[i]);
    }
    for(int i=0;i<n-1;i++){
        sort(bt+2*i,bt+n+i,cmp);
        T = cTree(bt[2*i],bt[2*i+1]);
        bt[n+i] = T;
    }
    Pre(T, "");
    printf("original:");
    decode(T,dc);
    return 0;
}