POJ 1521 Entropy

求哈夫曼树的平均码长，用优先队列来写，先记录某个字符在字符串里出现的次数，然后放入

队列。依次取出第一小和第二小的数，将两个数相加，构成新的虚拟结点，放入队列中。

/*Accepted    196K    0MS    C++    918B    2012-08-01 17:25:00*/

#include<stdio.h>

#include<string.h>

#include<stdlib.h>

#include<queue>

using namespace std;



int key[1 << 7], len;

char t[1 << 10];



int huffman()

{

    int i, a, b, c, sum = 0;

    priority_queue<int, vector<int>, greater<int> > q;

    for(i = 65; i < 100; i ++)

    {

        if(key[i] > 0)

            q.push(key[i]);

    }

    while(q.size() > 1)

    {

        a = q.top(); q.pop();

        b = q.top(); q.pop();

        c = a + b;

        sum += c;

        q.push(c);

    }

    return sum > 0 ? sum : len; //当只有一个字符的时候

}



int main()

{

    int i, sum;

    while(scanf("%s", t) != EOF)

    {

        if(strcmp(t, "END") == 0) break;

        len = strlen(t);

        memset(key, 0, sizeof key);

        for(i = 0; i < len; i ++)

            key[t[i]] ++;

        sum = huffman();

        printf("%d %d %.1f\n", len * 8, sum, (double)len * 8 / sum);

    }

    return 0;

}