http://acm.hdu.edu.cn/showproblem.php?pid=2527

Safe Or Unsafe

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

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  

#include<stdio.h>
#include<string.h>
#include<malloc.h>
#define MAX 10000
#define fun(c) (c-'a'+1)
struct HTnode
{//哈弗曼树的节点 
	int weight,parent; 
	int lchild,rchild;
	bool judge;
};
int htnum[27]/*huffmantree中的节点的权值*/;
char roadsize[27];//各节点的路径长 
void HuffmanTree(int n)//叶子节点个数
{
	if(n<=1) return;//如果只有一个节点
	
	//-----初始化树-----// 
    int m=2*n-1,i,j;
    struct HTnode *s;
    s=(struct HTnode *)malloc((m+1)*sizeof(struct HTnode));
	for(i=1;i<=n;i++)//初始化叶子节点
	{
		s[i].weight=htnum[i-1];
		s[i].parent=0;s[i].lchild=0;s[i].rchild=0;
		s[i].judge=false; 
	} 
	for(;i<=m;i++)//对除叶子节点的其他节点初始化 
	{
		s[i].weight=0;
		s[i].parent=0;
		s[i].lchild=0;
		s[i].rchild=0;
		s[i].judge=false;
	}
	//------------------//
	
	//-----构建huffmantree-----// 
	int s1,s2,min;
	for(i=n+1;i<=m;i++)
	{//每一次为节点1~i-1找两个孩子。这两个孩子是1~i-1中权值最小的，且没有父亲 
	    for(j=1,min=MAX;j<i;j++)
		if(s[j].weight<min&&!s[j].judge)
		{
		    min=s[j].weight;
		    s1=j;
		}
		s[s1].judge=true;
		for(j=1,min=MAX;j<i;j++)
		if(s[j].weight<min&&!s[j].judge)
		{
		    min=s[j].weight;
		    s2=j;
		}
		s[s2].judge=true;
		
		s[s1].parent=i;s[s2].parent=i;
		s[i].lchild=s1;s[i].rchild=s2;
		s[i].weight=s[s1].weight+s[s2].weight;
	}//huffmantree构建成功 
	//-------------------------//
	
	
	//-----计算叶子节点的路径长-----//
	memset(roadsize,0,sizeof(roadsize));
	for(i=1;i<=n;i++)
	{
		j=i;
		while(s[j].parent!=0)
		{
		    roadsize[i]+=1;
			j=s[j].parent;
		}
    }
	//-----------------------------//
	 
} 
int main()
{
	int N,limit,sum,n,i,j;
	char  s[MAX];
	int num[27];
	scanf("%d",&N);
	while(N--)
	{
		scanf("%d",&limit);
		scanf("%s",s);
		n=strlen(s);
		memset(num,0,sizeof(num));
		for(i=0;i<n;i++)//num数组对应存放a,b,c...z的个数,num[1]存放a的个数，以此类推 
		num[fun(s[i])]+=1;
		for(i=0,j=0;i<=26;i++)
		if(num[i]!=0)
		htnum[j++]=num[i];
		
		HuffmanTree(j);//构建huffmantree
		
		for(i=1,sum=0,n=j;i<=n;i++)
		sum+=htnum[i-1]*roadsize[i];
		
		if(sum<=limit) 
		printf("yes\n");
		else
		printf("no\n"); 
	}
	return 0;
}