hdu 2527 Safe Or Unsafe（优先队列+哈夫曼树）

#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
struct node
{
int t;
friend bool operator<(node aa,node bb)
{
return aa.t>bb.t;
}
};
int main()
{
char s[10005];
int a[30];
int i,ln;
int T;
scanf("%d",&T);
while(T--)
{
int n;
scanf("%d",&n);
getchar();
gets(s);
ln=strlen(s);
memset(a,0,sizeof(a));
for(i=0;i<ln;i++)
a[s[i]-'a'+1]++;
priority_queue<node>q;
node cur;
ln=0;
for(i=1;i<=26;i++)
{
if(a[i])
{
cur.t=a[i];
q.push(cur);
ln++;
}
}
int sum;
sum=0;
if(ln==1)
{
cur=q.top();
sum+=cur.t;
}
else
{
i=0;
while(i<ln-1)
{
i++;
int x,y;
x=q.top().t;
q.pop();
y=q.top().t;
q.pop();
cur.t=x+y;
sum+=x+y;
q.push(cur);
}
}
if(sum>n)
printf("no\n");
else
printf("yes\n");
}
return 0;
}