Safe Or Unsafe

Problem Description

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

Input

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

 

Output

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

#include<iostream>
#include<string>
#include<cstring>
using namespace std;
void swap(int &a,int &b);
void put(int heap[],int num,int &len);
int get(int heap[],int &len);
int main()
{
    int n,m,len,i,len1,len2,a,b;
    int num[26];
    int heap[40];
    int toal;
    string str;
    cin>>n;
    while(n--)
    {
        cin>>m;
        cin>>str;
        len=0;
        toal=0;
        memset(num,0,sizeof(num));
        len1=str.length();
        for(i=0;i<len1;++i)
        num[str[i]-'a']++;
        for(i=0;i<26;++i)
        if(num[i])
        put(heap,num[i],len);
        len2=len;
        if(len==1)
        toal=get(heap,len);
        else
        {
            for(i=0;i<len2-1;++i)
            {
                a=get(heap,len);
                b=get(heap,len);
                toal+=a+b;
                put(heap,a+b,len);
            }
        }
        if(toal<=m)
        cout<<"yes"<<endl;
        else
        cout<<"no"<<endl;
    }
}
void swap(int &a,int &b)
{
    int temp;
    temp=a;
    a=b;
    b=temp;
}
void put(int heap[],int num,int &len)
{
    ++len;
    heap[len]=num;
    int son=len;
    while(son!=1&&heap[son]<heap[son/2])

{
    swap(heap[son],heap[son/2]);

son/=2;

}
}
int get(int heap[],int &len)
{
    int min=heap[1];
    heap[1]=heap[len];
    --len;
    int son=1;
    while(son*2<=len)
    {
        if(son*2+1>len||heap[son*2]<heap[son*2+1])
        son=son*2;
        else
        son=son*2+1;
        if(heap[son/2]>heap[son])
        swap(heap[son/2],heap[son]);
        else
        break;
    }
    return min;
}