学步园

Javac++ 一天在看计算机的书籍的时候，看到了一个有趣的东西！每一串字符都可以被编码成一些数字来储存信息，但是不同的编码方式得到的储存空间是不一样的！并且当储存空间大于一定的值的时候是不安全的！所以Javac++ 就想是否有一种方式是可以得到字符编码最小的空间值！显然这是可以的，因为书上有这一块内容--哈夫曼编码(Huffman Coding)；一个字母的权值等于该字母在字符串中出现的频率。所以Javac++ 想让你帮忙，给你安全数值和一串字符串，并让你判断这个字符串是否是安全的？

输入有多组case，首先是一个数字n表示有n组数据，然后每一组数据是有一个数值m(integer)，和一串字符串没有空格只有包含小写字母组成！

如果字符串的编码值小于等于给定的值则输出yes，否则输出no。

2
12
helloworld
66
ithinkyoucandoit

no
yes
先建立一棵哈夫曼树，求除了最底层的值不用加外，其他的都加起来max，max与输入的值比较，大则输出no,否则输出yes,注意一种情况，当只有一层时，则直接比较。
#include<stdio.h>
#include<string.h>
#include<malloc.h>
#define INF 9999999
typedef struct nn
{
    int w;//权值
    int pare,lchilde,rchilde;
    int bm,b;//bm是编码：左0，右1；b判断当前节点是否被用过
}hafuman,*Hafuman;
int n;
Hafuman HF;
void vist(int &k1,int &k2)//求出最小的两个权值，位置以k1,k2反回
{
    int min1,min2;
    min1=min2=INF;
    for(int i=1;i<=n;i++)
    if(!HF[i].b)
        if(min1>HF[i].w) min1=HF[i].w,k1=i;
        else if(min2>HF[i].w) min2=HF[i].w,k2=i;
    if(k1>0) HF[k1].b=1;//己经用过
    if(k2>0) HF[k2].b=1;
}
void sethafuman(int m,int hs[])
{
    HF=(Hafuman)malloc(m*sizeof(hafuman));//开空间
    n=0;
    for(int i=1;i<27;i++)//建立最底层
     if(hs[i])
     {
         HF[++n].w=hs[i]; HF[n].b=0;
         HF[n].lchilde=-1;
         HF[n].rchilde=-1;
     }

    int k1,k2;
    while(1)//建立哈夫曼树
    {
        k1=k2=-1;
        vist(k1,k2);//在1~n中找出两个最小值的位置
        if(k1<0||k2<0) break;//只找到一个则己建好
        HF[++n].w=HF[k1].w+HF[k2].w;
        HF[n].b=0;
        HF[n].lchilde=k1;
        HF[n].rchilde=k2;
        
        HF[k1].bm=0; HF[k1].pare=n;
        HF[k2].bm=1; HF[k2].pare=n;
    }
}
int max;
void find(int v)
{
    if(v==-1) return ;
    int p=HF[v].pare;
    max+=HF[p].w; HF[p].w=0;//变0防止重复加
    find(HF[v].lchilde);
    find(HF[v].rchilde);
}
int main()
{
    int t,num,hs[30];
    char str[100000];
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %s",&num,str);
        memset(hs,0,sizeof(hs));
        int len=strlen(str),m=0;
        for(int i=0;i<len;i++)
        {
            if(hs[str[i]-'a'+1]==0) m++;//记录不同的字符个数
            hs[str[i]-'a'+1]++;
        }
        sethafuman(2*m+5,hs);
        if(m==1)//只有一层
        {
            if(HF[n].w>num) printf("no\n");
            else printf("yes\n");
            continue;
        }
        max=0;
        find(HF[n].lchilde);
        find(HF[n].rchilde);

        if(max>num) printf("no\n");
        else printf("yes\n");
    }
}