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