Safe Or Unsafe hdu2527

    哈夫曼树的模板题。

 

#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;
}