HDU 1053 [Entropy] Huffman编码

Posted 2020-10-04 哇咔咔咔

题目链接：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;
}