POJ1521 最优哈夫曼编码树 贪心算法的有效应用

题目链接：http://poj.org/problem?id=1521

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string.h>
#include<queue>
using namespace std;
const int maxn = 1e7+5;
char s[maxn];
int main()
{
    while (scanf("%s",s)&&strcmp(s,"END"))
    {
        int t=1;
        int len=strlen(s);
        sort(s,s+len);
        priority_queue<int ,vector<int> ,greater<int> > q;//将每个字符出现的频次放入优先队列 
        for(int i = 1 ; i < len ; i++)
        {
            if(s[i-1]!=s[i])
            {
                q.push(t);
                t=1;
            }
            else t++;
        }
        q.push(t);
        int ans=0;
        while(q.size()>1)//最终队列中只会剩下一个根结点的值，不必算在内 
        {
            int a=q.top();q.pop();
            int b=q.top();q.pop();
            ans+=a+b;//每次只加a+b就能得到频次*深度的原因是深度为dep的频次在之后会被加到其他树上，会多次计算 
            q.push(a+b); 
        }
        if(ans==0)ans=q.top();
        q.pop();
        int ans1=len*8;
        double ans3=(double)ans1/ans;
        printf("%d %d %.1lf
",ans1,ans,ans3); 
    }
 }