北邮考研复试的一道真题,借助动态数组解决求解哈夫曼树的权值问题的方法。之前自己也没有接触过此类用法,整理下来,希望对各位有帮助。
题目描述
哈夫曼树,第一行输入一个数n,表示叶结点的个数。需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即weight,题目需要输出所有结点的值与权值的乘积之和的最小值。
输入描述:
输入有多组数据。
每组第一行输入一个数n,接着输入n个叶节点(叶节点权值不超过100,2<=n<=1000)。
输出描述:
输出权值。
示例1
输入
5
1 2 2 5 9
输出
37
问题分析
解决这道题的关键是要理解哈夫曼树的构造过程,基本思路如下(假设给出一个含有n个元素的数组):
从数组中取出最小的两个元素进行求和,将求和后的新元素插入到数组,此时,数组元素减少1个,如果数组元素仅剩1个,构造结束,否则重复上述过程至仅剩一个数组元素。
上述过程中,每次加入的初始元素为叶结点。
下面以示例1的数据为基础进行分析哈夫曼树构造过程:
1、从1,2,2,5,9所组成的数组中,取出最小的两个数值1,2加和得到3,将3插入数组,原数组变为2,3,5,9。
2、从2,3,5,9中取出最小的两个数据2,3加和得到5,将5插入数组,数组变为5,5,9。
3、从5,5,9中取出5,5加和得到10,将10插入数组,数组变为9,10。
4、从9,10中取出9,10,加和得到19,插入数组,此时,数组仅剩一个元素19结束。
求解哈夫曼树的权值,仅需将权值加入到上述1-4过程。在1中1,2首先作为叶结点,深度为1,此时,权值增加weight1=(1+2)*1=3。在2中选取2,3,将它们看做叶结点,深度为1,此时权值增加weight2=(weight1+2)*1=5。在3中选择5,5,因此,将5,5作为叶结点,深度为1,权值增加weight3=(weight2+5)*1=10。在4中取出9,10,看做叶结点,深度为1,权值增加weigt4=(weight3+9)*1=19。最终,权值weight=weight1+weight2+weight3+weight4=3+5+10+19=37
代码
#include<bits/stdc++.h>
using namespace std;
int main(){
int n,t;
cin>>n;
vector<int>v;
for(int i=0;i<n;i++){
cin>>t;
v.push_back(t);
}
sort(v.begin(),v.end());
int weight=0;
while(v.size()!=1){
int a=*(v.begin());
v.erase(v.begin());
int b=*(v.begin());
v.erase(v.begin());
weight=weight+a+b;
v.push_back(a+b);
sort(v.begin(),v.end());
}
cout<<weight;
return 0;
}
题目地址:哈夫曼树