hdu2527哈夫曼树

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

解题思路：哈夫曼树裸题，方法就是使用优先队列维护（找出权重第一小和第二小的两个元素，然后合并成一个放入队列中，复杂度为O(nlogn）)，但题目也有莫名奇妙的地方，就是len=1和len>1时的计算方法略有不同。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
priority_queue<int,vector<int>,greater<int> > s;
char a[4000];
int p[30];
int n;
int main()
{
	//freopen("t.txt","r",stdin);
	int T,l;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		getchar();
		scanf("%s",a);
		l=strlen(a);
		memset(p,0,sizeof(p));
		int len=0;
		for(int i=0;i<l;i++)
		{
			p[a[i]-'a']++;
		}
		while(!s.empty()) s.pop(); 
		for(int i=0;i<26;i++)
		{
			if(p[i]!=0)
			{
				s.push(p[i]);
				len++;
			}
		}
		int ans=0;
		if(len==1)
		{
			ans=s.top();
			s.pop();
		}else
		{
			while(len>1)
			{
				int min1=s.top();
				s.pop();
				int min2=s.top();
				s.pop();
				ans+=(min1+min2);
				s.push(min1+min2);
				len--;
			}
			s.pop();
		}
		//printf("%d\n",ans);
		if(ans<=n) printf("yes\n");
		else printf("no\n"); 
	}
	return 0;
}