树的一些概念:
路径:从树中任一结点到达另一结点的通路称为路径
路径长度:路径上所经过的边的个数称为路径长度
结点权值:树结点带有某种意义的数值称为权值
结点的带权路径长度:从树根结点到树中某结点的路径长度乘以该结点的权值的值
树的带权路径长度:所有叶结点的带权路径长度之和
哈夫曼树:树的带权路径长度最小的二叉树(这里我们默认是二叉哈夫曼树),也称为最优树
哈夫曼树的一些应用:求哈弗曼编码,多个数两两合并时的最小比较次数等
如何求哈夫曼树:
- 将所有结点放入集合K中
- 若集合K中结点个数大于2,取其中两个权值最小结点合并得到一新结点,两权值相加得到新值保存在新结点,新结点重新放回集合K。重复2,3步
- 若集合中只剩一个结点,该结点即为构造出的哈夫曼树的根结点。
注意:所有构造得到的中间结点(哈夫曼树上的非叶结点)的权值之和等于该哈夫曼树的带全路径长度
上面求哈夫曼树过程中会不断取集合中权值最小的两个结点,可以使用堆数据结构中的小顶堆,它可以以O(log2n)的复杂度
取得n个数中的最小元素为了绕过对堆得实现,可使用STL中的标准模板——优先队列
STL中的优先队列默认为大顶堆
priority_queue<int> Q1; //定义一个大顶堆
对其进行重构:
priority_queue<int,vector<int>,greater<int>> Q; //定义一个小顶堆
优先队列的基本操作:
Q.push(x);//进队
int a = Q.top();//取堆顶元素,保存于a中
Q.pop();//弹出堆顶元素
预处理:
#include<queue>
#include<functional>
using namespace std;
题目描述:
哈夫曼树,第一行输入一个数n,表示叶结点数,第二行输入n个叶结点的权值,权值jian用空格隔开。用这些叶结点构造
哈夫曼树,输出哈夫曼树的带权路径长度。
样例输入:
6
1 3 5 2 6 8
样例输出:
59
直接看代码吧:
#include<stdio.h>
#include<queue>
#include<functional>
using namespace std;
priority_queue<int,vector<int>,greater<int>> Q; //定义一个小顶堆
//重载后的优先队列可以自动将队列排成小顶堆
int main()
{
int n,x;
int a,b;
int ans;
while (scanf("%d",&n)!=EOF)
{
while (!Q.empty()) Q.pop(); //清空队列
ans=0;
for (int i = 0; i < n; i++) //输入叶节点权值并建立小顶堆(优先队列自动建立)
{
scanf("%d",&x);
Q.push(x);
}
while (Q.size() >1)
{
a = Q.top();Q.pop(); //取集合(小顶堆)中最小的两个权值
b = Q.top();Q.pop();
ans+=a+b; //得到两个叶子节点父节点的权值
Q.push(a+b); //将父节点的权值并入集合中
}
printf("%d\n",ans);
}
return 0;
}
黑框框运行结果: