BUPT复试专题—哈夫曼编码(2009)

题目描述

哈夫曼编码中  平均码长=码长×码字出现的概率

如：ABCDE 五个字符的出现次数分别为50 20 5 10 15

那么，其哈夫曼编码为A:0   B:10   C:1110   D:1111   E:110

该哈夫曼编码的平均码长=(50*1+20*2+5*4+10*4+15*3)/100=1.95

输入

有多组输入，每组两行

第一行:字符的个数 N

第二行:N 个以空格隔开的数,表示这 N 个字符中每个字符出现次数

输出

输出该哈夫曼编码的平均码长，保留两位小数

样例输入

5
50 20 5 10 15

样例输出

1.95

来源

2009机考D题

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
int a[];
char s[];
struct node
{
    int w;
    friend bool operator <(node aa, node bb) //<为从大到小排列，>为从小到大排列
    {
        return aa.w > bb.w;
    }
};
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(a, , sizeof(a));
        int num=,len=n;
        for(int i=;i<=n;i++) //题目已知每一种字母有多少个，没已知要自己数
            cin>>a[i];
        priority_queue <node> q;
        for(int i=;i<;i++)
        {
            num+=a[i];
            node b;
            b.w=a[i];
            if(a[i])
                q.push(b);
        }
        int res;
        if(q.size() == )
            res = len;
        else
        {
            res = ;
            while(q.size() > )
            {
                int aa = q.top().w; q.pop();
                int bb = q.top().w; q.pop();
                res += (aa + bb);
                node b;
                b.w = aa + bb;
                q.push(b);
            }
        }
        double temp=double(res)/(double)num;
        printf("%.2lf\n",temp);
    }
    return ;
}