HDU -2527 -- Safe Or Unsafe【哈夫曼编码模板题】

Safe Or Unsafe

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

思路

统计字符串出现的次数，然后就是一个哈夫曼树（wpl）纯模板题。
坑点1：特殊判断如果全是一个字符的字符串，因为你会发现进入不了求wpl的过程
坑点2：本题是多组输入，而每组测试样例的队列都没有将哈夫曼树的根节点删除。
建立良好的书写代码习惯不管是否是多组数据都应该删除一下。

AC代码

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
int arr[30];
priority_queue<int,vector<int>,greater<int> >q;
int main(){
	string s;
	int t,m;
	cin>>t;
	while(t--){
		cin>>m>>s;
		int len=s.size(),wpl=0;
		for(int i=0;i<26;i++)	arr[i]=0;
		for(int i=0;i<len;i++)	arr[s[i]-'a']++;
		for(int i=0;i<26;i++)	if(arr[i]!=0)	q.push(arr[i]);
		if(q.size()==1)
			wpl=q.top();
		else{
			while(q.size()>1){
				int min1=q.top();q.pop();
				int min2=q.top();q.pop();
				wpl+=min1+min2;
				q.push(min1+min2);
			}
		}
		q.pop();//删除顶端这个元素 
		if(wpl<=m)	puts("yes");
		else	puts("no");
	}
	return 0;
}