HDU 2527 Safe Or Unsafe 哈夫曼树

来源：互联网发布：esp8266单片机程序编辑：程序博客网时间：2024/05/01 14:51

题目地址：　http://acm.hdu.edu.cn/showproblem.php?pid=2527

明天蓝桥杯比赛，看了一下以前做的题目，发现好多都要看下以前写的才能ＡＣ，哎。。。。

一道基本的哈夫曼树问题，直接用优先队列模拟。

代码入下：

#include<iostream>#include<vector>#include<list>#include<deque>#include<queue>#include<stack>#include<map>#include<set>#include<algorithm>#include<cstdio>#include<cstdlib>#include<cstring>#include<string>#include<cmath>using namespace std;typedef long long LL;const int N=101000;char s[N];int a[27];struct xh{    int t;    friend bool operator<(xh aa,xh bb)    {        return aa.t>bb.t;    }}w;int main(){    int i,j,len,T,sum,n;    cin>>T;    while(T--)    {        scanf("%d%s",&n,s);        len=strlen(s);        memset(a,0,sizeof(a));        for(i=0;i<len;i++)            a[s[i]-'a']++;        priority_queue<xh> q;        len=0;        for(i=0;i<26;i++)            if(a[i])            {                w.t=a[i];                q.push(w);                len++;            }        if(len==1)        {            sum=q.top().t;        }        else        {            int i=0;            sum=0;            while(i<len-1)            {                int x=q.top().t;                q.pop();                int y=q.top().t;                q.pop();                sum+=x+y;                w.t=x+y;                q.push(w);                i++;            }        }        if(sum<=n)            printf("yes\n");        else            printf("no\n");    }    return 0;}