1058: 哈夫曼
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 14 Solved: 10
Submit Status Discuss
Description
1111这一天,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 "cstring"
#include "cstdio"
#include "string.h"
#include "iostream"
#include "queue"
#include "vector"
#include "algorithm"
using namespace std;
struct cmp //优先队列需要的排序函数,升序
{
bool operator()(int &a,int &b)
{
return a>b;
}
};
int main()
{
int ncase;
scanf("%d",&ncase);
while(ncase--)
{
int max;
scanf("%d",&max);
char a[1500];//字符串
//int cnt[27];//
int aa[27];
//memset(cnt,0,sizeof(cnt));
memset(aa,0,sizeof(aa));
scanf("%s",a);
for(int i=0;;i++)
{
if(a[i]=='\0')
break;
aa[a[i]-'a']++;
}
priority_queue<int,vector<int>,cmp> list;
for(int i=0;i<=25;i++)
{
if(aa[i]!=0)
list.push(aa[i]);
}
int sum=0;
while(!list.empty()&&list.size()!=1)
{
int temp1=list.top();
list.pop();
int temp2=list.top();
list.pop();
int temp3;
temp3=temp2+temp1;
list.push(temp3);
sum+=temp3;
}
if(sum<=max)
printf("yes\n");
else
printf("no\n");
}
}