uvalive 2088 - Entropy(huffman编码）

题目连接：2088 - Entropy

题目大意：给出一个字符串， 包括A~Z和_， 现在要根据字符出现的频率为他们进行编码，要求编码后字节最小， 然后输出字符均为8字节表示时的总字节数， 以及最小的编码方式所需的总字节数，并输出两者的比率， 保留一位小数。

解题思路：huffman编码。

#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;
const int N = 10005;

struct Node {
int r;
int l;
int cnt;
bool operator < (const Node& a) const {
return cnt > a.cnt;
}
}tmp
;

int len, sum, vis
;

void init(char str[]) {
memset(tmp, 0, sizeof(tmp));
memset(vis, 0, sizeof(vis));
sum = 0;
len = strlen(str);

for (int i = 0; i < len; i++) {
if (str[i] == '_')
vis[26]++;
else
vis[str[i] - 'A']++;
}
}

void count(Node cur, int deep) {
if (cur.l == -1 && cur.r == -1) {
sum += deep * cur.cnt;
return;
}
if (cur.l != -1)
count(tmp[cur.l], deep + 1);
if (cur.r != -1)
count(tmp[cur.r], deep + 1);
}

void solve() {
int n = 0;
Node now;
priority_queue<Node> que;
for (int i = 0; i < 27; i++)
if (vis[i]) {
now.cnt = vis[i];
now.l = now.r = -1;
que.push(now);
}

while (1) {
tmp[n++] = que.top(), que.pop();
if (que.empty()) break;
tmp[n++] = que.top(), que.pop();
now.cnt = tmp[n - 1].cnt + tmp[n - 2].cnt;
now.l = n - 1, now.r = n - 2;
que.push(now);
}
if (n == 1)
sum = tmp[n - 1].cnt;
else
count(tmp[n - 1], 0);
}

int main() {
char str
;
while (scanf("%s", str) == 1) {
if (strcmp(str, "END") == 0) break;

init(str);
solve();
printf("%d %d %.1lf\n", len * 8, sum, len * 8.0 / sum);
}
return 0;
}