4080:Huffman编码树
总时间限制: 1000ms 内存限制: 65536kB
描述
构造一个具有n个外部节点的扩充二叉树,每个外部节点Ki有一个Wi对应,作为该外部节点的权。使得这个扩充二叉树的叶节点带权外部路径长度总和最小:
Min( W1 * L1 + W2 * L2 + W3 * L3 + … + Wn * Ln)
Wi:每个节点的权值。
Li:根节点到第i个外部叶子节点的距离。
编程计算最小外部路径长度总和。
输入
第一行输入一个整数n,外部节点的个数。第二行输入n个整数,代表各个外部节点的权值。
2<=N<=100
输出
输出最小外部路径长度总和。
样例输入
4
1 1 3 5
样例输出
17
与此题无关的Huffman树的构造方法:还是使用最小堆,但是每次push进去的是node,这样队列中最后留下的q.top就是树的root了。
#include<iostream>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
//http://bailian.openjudge.cn/practice/4080/
//解锁新知识,因为路径长度都是单位距离
//huffman外部路径长度和 就是每个小分支的左右节点相加的和
// 也就是内部节点加和:2+5+10
priority_queue<int,vector<int>,greater<int> > q;
int n,a,res=0;
int main(){
cin>>n;
for(int i=0;i<n;i++){
cin>>a;
q.push(a);
}
while(q.size()>1){
int x=q.top();
q.pop();
int y=q.top();
q.pop();
int t=x+y;
res=res+t;
q.push(t);
}
cout<<res<<endl;
}