1022-哈夫曼编码与译码

描述

已知电文包括的字符集为{A，C，I，M，N，P，T，U}，输入对应权值，对字符集合进行哈夫曼编码，完成电文的哈夫曼编码与译码工作。

输入

 

共三行：

第一行为对应字符集{A，C，I，M，N，P，T，U}的权值

第二行为一段字符串表示的电文(长度不超过1000）；

第三行为一段电文的哈夫曼编码。

输出

 

共十行：

前八行为各字符的编码；

第九行是与第二行输入对应的哈夫曼编码；

第十行是与第三行输入对应的电文。

样例输入

1 2 3 4 5 6 7 8

NUPTICPCACM

1111011111100

样例输出

A: 11110

C: 11111

I: 1110

M: 100

N: 101

P: 110

T: 00

U: 01

1010111000111011111110111111111011111100

ACM

#include<iostream>
#include<queue>
#include<string>
using namespace std;
#define MAX 10


char str[8]={'A','C','I','M','N','P','T','U'};
int temp[MAX*3];
int str2[256][MAX*3];
string str3;

class node
{
    public:
    int k;
    char c;
    node *l,*r;
    node()
    {
        c='0';
        k=0;r=l=NULL;
    }
    node(int a,char s)
    {
        k=a;r=l=NULL;
        c=s;
    }
    friend node* uni(node *p,node *q)
    {
        node *temp=new node(p->k+q->k,'0');
        temp->l=p;
        temp->r=q;
        return temp;
    }
};
void prim(node *p,int L)
{
    if(!p->r)
    {
        str2[p->c][0]=L;
        for(int i=1;i<=L;i++)
        {
            str2[p->c][i]=temp[i-1];
        }
    }
    else
    {
        temp[L]=0;
        prim(p->l,L+1);
        temp[L]=1;
        prim(p->r,L+1);
    }
}

class nodeCmp
{
public:
    bool operator()(node* p1, node* p2) const
    {
        return p1->k > p2->k;
    }
};
node *s[MAX*3];
priority_queue<node *,deque<node *>,nodeCmp> pro_queue;
int main()
{
    //freopen("a.txt","r",stdin);
    int i,a;
    for(i=0;i<8;i++)
    {
        cin>>a;
        node *p;
        p=new node(a,str[i]);
        s[i]=p;
        pro_queue.push(s[i]);
    }
    while(pro_queue.size()>1)
    {
        node *p,*q,*k;
        p=pro_queue.top();
        pro_queue.pop();
        q=pro_queue.top();
        pro_queue.pop();
        k=uni(p,q);
        pro_queue.push(k);
        }
        node *p=pro_queue.top();
        prim(p,0);
        for(i=0;i<8;i++)
        {
            cout<<str[i]<<": ";
            for(int j=1;j<=str2[str[i]][0];j++)
            {
                cout<<str2[str[i]][j];
                }
                cout<<endl;
            }
            cin>>str3;
            int len=str3.length();
            for(i=0;i<len;i++)
            {
                for(int j=1;j<=str2[str3[i]][0];j++)
                {
                cout<<str2[str3[i]][j];
                }
            }
            cout<<endl;
            cin>>str3;
            len=str3.length();
            node *q=pro_queue.top();
            for(i=0;i<len;i++)
            {

                if(str3[i]=='1')
                {
                q=q->r;
                }
                else
                {
                q=q->l;
                }
                if(q->c!='0')
                {
                cout<<q->c;
                q=pro_queue.top();
                }
            }
            cout<<endl;
    return 0;
    }