Pat 04-树6. Huffman Codes (30)

题目链接：

Huffman codes

 

 

 

题意：

先给出N个节点的出现次数

再给出M种编码方式 

判断每种编码方式是否能构成哈夫曼树

 

 

 

题解：

判断哈夫曼编码的条件有两个：

1  哈夫曼编码不唯一，但它的WPL(带权路径长度)一定唯一

2  短码不能是长码的前缀

 

首先可以使用STL优先队列 根据  WPL=所有非叶节点的权值之和   求出标准的WPL1

再根据WPL2=所有叶节点的高度*权值之和

再单独判断是否编码中构成前缀

两个条件都满足则输出Yes

#include
#include
#include
#include
#include
using namespace std;
struct node1{
    char ch;
    int num;
}aa[105];
struct node2{
    char ch;
    string str;
}bb[105];
int n;
int cmp(node2 a,node2 b){
    return a.str.size(),greater >q;
    for(int i=1;i<=n;i++)
    {
        cin>>aa[i].ch>>aa[i].num;
        q.push(aa[i].num);
    }
    int a,b;
    while(!q.empty()){
        a=q.top();
        q.pop();
        if(!q.empty()){    //最后队列中只会剩下一个根节点
            b=q.top();
            q.pop();
            q.push(a+b);
        }
        ans1+=a+b;
    }
    ans1=ans1-a-b;      //  WPL=所有非叶节点的权值之和  
    scanf("%d",&m);
    while(m--){
        ans2=0;
        for(int i=1;i<=n;i++){
            cin>>bb[i].ch>>bb[i].str;
            ans2+=aa[pos(bb[i].ch)].num*bb[i].str.size(); //WPL2=所有叶节点的高度*权值之和
        }
        if(ans2!=ans1)
            cout<<"No"<