题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1053
题目大意:给一个字符串(只包含A-Z和下划线_表示空格),问8bit的ASCII编码和最优编码所需位数,并求比值。
关键思想:先统计权值,前者就是8*长度,后者就是让我们求Huffman编码的带权路径长度,我们可以用优先队列实现。特殊情况要考虑一下。
代码如下:
#include <iostream> #include <algorithm> #include <string.h> #include <queue> using namespace std; bool cmp(int a,int b){ return a>b; } int Huff(int c[],int n){ if(n==1)return c[0];//只有一个权值就返回该权值 int ans=0; priority_queue<int,vector<int>,greater<int> >q;//小的在队列首 for(int i=0;i<n;i++)q.push(c[i]); while(q.size()>=2){//要合并两个结点 int f=q.top();q.pop(); int s=q.top();q.pop(); q.push(f+s); ans+=(f+s); } return ans; } int main(){ string a; int c[27]; while(cin>>a&&a!="END"){ memset(c,0,sizeof(c)); for(int i=0;i<a.size();i++){ if(a[i]!='_')c[a[i]-'A']++; else c[26]++; }//统计权值 sort(c,c+27,cmp); int cnt=0; while(c[cnt]!=0)cnt++;//共有几种相异字符 int huff=Huff(c,cnt); printf("%d %d %.1f\n",a.size()*8,huff,a.size()*8.0/huff); } return 0; }