Safe Or Unsafe

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 1578    Accepted Submission(s): 617

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

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

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

Sample Input

2 12 helloworld 66 ithinkyoucandoit  

Sample Output

no yes  

Source
HDU 2008-10 Programming Contest 

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
char s[100000];
struct node
{
	int data;
	int parent;
	int lchild;
	int rchild;
}huffman[1000];
int main()
{
	int m,a[27],b[27],c[27];
	int n,i,j;
	scanf("%d",&n);
	while(n--)
	{
		int s1=0;
		memset(a,0,sizeof(a));
		memset(b,0,sizeof(b));
		memset(c,0,sizeof(c));
		scanf("%d",&m);
		getchar();
		scanf("%s",s);
		int len=strlen(s);
		for(i=0;i<len;i++)
		{
			a[s[i]-'a']++;//a数组存放字符串中各个字符出现的次数
		}
		int k=0;
		for(i=1;i<=26;i++)
		{
			if(a[i])
			b[++k]=a[i];//b数组存放的是huffman树中叶子节点的权值 
		}
		if(k!=1)//k==1时，也就是叶子节点只有一个，无法构成huffman树 
		{//叶子节点数目不为1是，就开始构造huffman树 
			memset(huffman,0,sizeof(huffman));
			for(i=1;i<=k;i++)
			{
				huffman[i].data=b[i];
			}
			for(i=1;i<k;i++)//原有k个叶子节点，需要补充k-1个节点 
			{					//因为huffman树有2*k-1个节点 
				int max=10000000;
				int m1,m2,x1,x2;
				m1=max;
				m2=max;//m2存上一个m1的值,m1存最新的值,x2存上一个x1的值,x1存最新的值
				x1=0;//m1,m2存结点的权值,x1,x2分别存结点的左右孩子的位置
				x2=0;
				for(int j=1;j<k+i;j++)
				{
					if(huffman[j].data<m1&&huffman[j].parent==0)
					{
						m2=m1;
						m1=huffman[j].data;
						x2=x1;
						x1=j;
					}
					else if(huffman[j].data<m2&&huffman[j].parent==0)
					{
						m2=huffman[j].data;
						x2=j;
					}
				}
				huffman[k+i].lchild=x1;
				huffman[k+i].rchild=x2;
				huffman[x1].parent=k+i;
				huffman[x2].parent=k+i;
				huffman[k+i].data=huffman[x1].data+huffman[x2].data;
			}
			//测试输出huffman树 
			/**for(i=1;i<2*k;i++)
			{
				printf("%d %d %d %d %d",i,huffman[i].data,huffman[i].lchild,huffman[i].rchild,huffman[i].parent);
			}*/
			int h;
			for(i=1;i<=k;i++)
			{
				j=i;
				for(;;)//这里循环条件是空，所以下面要判断h==2*k-1  否则会死循环 
				{
					h=huffman[j].parent;
					j=h;
					c[i]++;
					if(h==2*k-1) break;
				}
				s1+=b[i]*c[i];
			}
		}
		else
		s1=b[1]*1;
		if(s1<=m) printf("yes\n");
		else printf("no\n");
	}
	return 0;
}