哈夫曼树的模板题。
#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
struct node
{
int w;
node *left,*right;
};
int size;
node *root,static_memory[1000000];
int ans;
struct cmp
{
bool operator()( int a, int b)
{
return static_memory[a].w>static_memory[b].w;
}
};
void build(int p[])
{
size=0;
priority_queue<int,vector<int>,cmp>q;
for(int i=0;i<26;i++)
if(p[i])
{
node *ptr=static_memory+size;
ptr->w=p[i];
ptr->left=ptr->right=NULL;
q.push(size++);
}
if(q.size()==1)
{
ans=static_memory[0].w;
root=static_memory;
return;
}
while(!q.empty())
{
node *p1=static_memory+q.top();
q.pop();
node *p2=static_memory+q.top();
q.pop();
node *p=static_memory+size;
p->w=p1->w+p2->w;
p->left=p1;
p->right=p2;
root=p;
if(q.empty())
break;
q.push(size++);
}
}
void list_tree(node *p,int depth)
{
if(p->left==NULL&&p->right==NULL)
{
ans+=p->w*depth;
return;
}
list_tree(p->left,depth+1);
list_tree(p->right,depth+1);
}
int main()
{
int t,n,p[26],i;
char str[10000];
scanf("%d",&t);
while(t--)
{
scanf("%d%s",&n,str);
for(i=0;i<26;i++)
p[i]=0;
i=-1;
while(str[++i])
p[str[i]-'a']++;
ans=0;
build(p);
list_tree(root,0);
if(ans<=n)
printf("yes\n");
else
printf("no\n");
}
return 0;
}