算法导论2 C 哈夫曼编码（哈弗曼树，贪心）

题目：

题目描述

给定一只含有小写字母的字符串；输出其哈夫曼编码的长度

输入

第一行一个整数T，代表样例的个数，接下来T行，每行一个字符串，0<T<=2000,字符串长度0<L<=1500.

输出

对于每个字符串，输出其哈夫曼编码长度

样例输入 Copy

3

hrvsh

lcxeasexdphiopd

mntflolfbtbpplahqolqykrqdnwdoq

样例输出 Copy

10

51

115

解析：

可能题目有比较强的迷惑性，诱导人建造哈弗曼树，建造哈夫曼编码，再从哈夫曼编码中一个个查某个字符所占位数。

实际上这道题要求的就是以题目中给出的字符组成的哈夫曼树的根节点的权值。（可能上课时用大量时间讲哈夫曼编码，让很多人忘记了哈弗曼树根节点权值的含义）

这样只需统计字符串中出现的字符和这些字符出现的次数，把这些字符当做叶子节点，字符出现的次数当做权值，自底向下计算出根节点的权值即可。

上述过程与作业1的锯木条一模一样。

代码：

#include<iostream>
#include<queue>
using namespace std;
int vis[26];//散列表记录每个字母出现的次数
void solve()//实际上就是求哈夫曼树的根节点的值罢了，与作业1锯木条一摸一样 
{
	for(int i=0;i<26;i++)
		vis[i] = 0;
	priority_queue<int,vector<int>,greater<int>> que;//小顶堆
	string a;
	cin>>a;
	int idx = 0;
	while(a[idx]!='\0')
		vis[a[idx++]-'a']++;
	int n = 0;
	for(int i=0;i<26;i++)
	{
		if(vis[i]!=0)
		{
			que.push(vis[i]);
			n++;
		}
	}
	int ans = 0;
	for(int i=0;i<n-1;i++)//哈夫曼树 2n-1个节点，n-1个非叶子节点 
	{
		int a,b;
		a = que.top();
		que.pop();
		b = que.top();
		que.pop();
		int t = a+b;
		ans+=t;
		que.push(t);
	}
	cout<<ans<<'\n';
}
int main()
{
	int t;
	cin>>t;
	while(t--)
		solve();
}