HDU 1053 [Entropy] 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;
}