HDU 1053 Entropy 模拟哈夫曼树

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
int num[27];
int fun()
{
int n = 0,i,j,l;
sort(num, num+27); // 按出现次数从小到大排序 也就是按频率升序排序
int tmp[27],k = 0;
for(i = 0; i < 27; i ++)
if(num[i] != 0)
break;
// index [0,26]
l = 27-i; // l 表示出现的字母个数
if(l == 1)
return num[26];
for(i = 27-l; i < 27; i ++)
tmp[k++] = num[i];

// 模拟哈夫曼的构造过程（l个点 只需要构建l-2次）
for(i = 0; i < l - 1; i ++) {
if(i%2==0){
num[0] = tmp[0] + tmp[1];
n += num[0]; // 累加长度
for(j=2; j<k; j++)
num[j-1] = tmp[j];
k--; // 所有点的个数-1 (2点已合1点)
sort(num, num+k);  // 组合一次就重新排序
}
else{
tmp[0] = num[0]+num[1];
n += tmp[0]; // 累加长度
for(j=2; j<k; j++)
tmp[j-1] = num[j];
k--;
sort(tmp, tmp+k);  // 组合一次就重新排序
}
}

return n;
}
char str[1001];
int main()
{
int len,i;
int original, now;
while(scanf("%s", str)&&strcmp(str,"END")!=0){
memset(num, 0, sizeof(num));
len = strlen(str);
for(i = 0; i < len; i ++) {
if(str[i]-'A'<26)
num[str[i]-'A'] ++;
else
num[26] ++; // 下划线
}
original = 8*len;
now = fun(); // 哈夫曼二进制压缩  返回压缩位数
printf("%d %d %.1lf\n", original, now, 1.0*original/now);
}
return 0;
}