题目链接哈夫曼树
题目是求所有结点值和权值的乘积之和,即叶节点的最小带权路径之和。结点值为树的路径长度,权值为给定值。
可以将所求值等价于累加选出的两个最小权值结点的权值,直到根节点,其中对根节点不再累加。整个过程就是哈夫曼树的从下到上的选点过程,即将除根节点外的所有结点累加得到结果值。
采用了优先级队列来实现最小堆的功能,简化了代码。
代码如下:
#include <iostream>
#include <queue>
using namespace std;
struct Node
{
int id;
int depthr;
};
int main()
{
priority_queue<int, vector<int>, greater<int> > p;
int n, x;
while(cin >> n)
{
for(int i=0; i<n; i++)
{
cin >> x;
p.push(x);
}
int ans = 0;
while(p.size()>1)
{
int a = p.top(); p.pop();
int b = p.top(); p.pop();
int c = a+b;
ans += c;
p.push(c);
}
cout << ans << endl;
while(!p.empty()) p.pop();
}
return 0;
}