HDOJ 1053 Entropy

链接：Problem - 1053 (hdu.edu.cn)

问题描述：

问题分析：

运用哈夫曼树的概念进行解题.

代码实现：

 

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <functional>
#include <string>
using namespace std;
string s;
int num[31];
int bfs()
{
    int a,b,sum=0;
    priority_queue<int ,vector<int>,greater<int> >q;
    for(int i=0;i<=26;i++)
    {
        if(num[i]) q.push(num[i]);
    }
    if(q.size()==1) sum=q.top();
    while(q.size()>1)
    {
        a=q.top();
        q.pop();
        b=q.top();
        q.pop();
        sum+=a+b;
        q.push(a+b);
    }
    return sum;
}

int main()
{
    while(cin>>s)
    {
        if(s=="END") break;
        memset(num,0,sizeof(num));
        int len=s.size();
        for(int i=0;i<len;i++)
            if(s[i]=='_')
                num[26]++;
            else
                num[s[i]-'A']++;
        int res =bfs();
        printf("%d %d %.1lf\n",len*8,res,len*8*1.0/res*1.0);
    }
    
    return 0;
}