Hduoj2527【哈夫曼树】

/*Safe Or Unsafe 
Time Limit : 2000/1000ms (Java/Other)   Memory Limit : 32768/32768K (Java/Other)
Total Submission(s) : 9   Accepted Submission(s) : 5
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

Source
HDU 2008-10 Programming Contest
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
char s[10000010];
int mark[27], n, m;
int cmp(const void *a, const void*b)
{
	return *(int *)a - *(int *)b;
}
void sort(int x)
{
	int i, j, k;
	k = mark[x];
	for(i = x+1; i < 26; ++i)
	{
		if(mark[i] >= k)
		{
			for(j = x; j < i; ++j)
			mark[j] = mark[j+1];
			mark[i] = k;
			return ;
		}
	}
	for(j = x; j < 25; ++j)
	mark[j] = mark[j+1];
	mark[25] = k;
}
int main()
{
	int i, j, k, t, ans;
	scanf("%d", &t);
	while(t--)
	{
		scanf("%d", &n);
		getchar();
		gets(s);
		m = strlen(s);
		memset(mark, 0, sizeof(mark));
		ans = 0;
		for(i = 0; i < m; ++i)
		mark[s[i] - 'a']++;
		qsort(mark, 26, sizeof(mark[0]), cmp);
		for(i = 0; i < 26; ++i)
		{
			if(mark[i] != 0)
			{
				for(j = i; j < 26; ++j)
				{
					mark[j+1] += mark[j];
					ans += mark[j+1];
					sort(j+1);
				}
				if(ans != mark[26])//处理当只有一种字符时的情况 
				ans -= mark[26];
				break;
			}
		}
		if(ans <= n)
		printf("yes\n");
		else
		printf("no\n");
	}
	return 0;
}

题意：给出一个字符串，字符串中的每种字母都有一个权值，其权值即为它在字符串中出现的频率，现在要求将每种字母的权值作为一个节点进行哈夫曼树的构造，算出树的权值是否小于等于安全值，若是输出yes，否则no。

思路：此题考查的就是对哈夫曼树的理解，刚开始忘了哈夫曼树的权值怎么算的了。。。一直错，所以一定要对哈夫曼树的定义非常清楚，树的权值就是非叶子节点的总和，所以我这里用贪心去做，即先对所有权值进行升序排序，每次从数组中取出2个最小的值相加，那么加出来的这个数肯定是非叶子节点加入ans中，其次加完之后将数组中前一个数直接遗弃，对于新加出来的数则作为加数的后一个，即数组中的后一个数，然后对该数进行排序插入操作，为的就是保证数组中前2个数一定是数组中所剩余的数是按升序排列的，然后只要一直加到底，并且每次都把加出来的和加入ans中即可，ans即为哈夫曼树的权值。其次我这里为了处理当字符串中只有一种字母的情况，将加数多扩了一位并且置零，若是只有一种字母，那么ans肯定等于数组中的最后一位，若不止一种字母，那么ans必定不等于数组中的最后一位。