题目所描述的有点问题,其实是叶节点的路径与权值乘积之和,即带权路径长度!
用到了c++中小顶堆的创建方法,优先队列。
定义:priority_queue<Type, Container, Functional>
Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆
还利用到:若定义每个结点都有值,为其子女的值之和,那么除了根结点外所有结点之和恰好等于带权路径长度,按层级累加的规律不难理解叶子节点的值也是“乘了好多倍”
#include<iostream>
#include<stdio.h>
#include<queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> > Q;
int main(){
int N;
while(scanf("%d",&N)!=EOF){
while(!Q.empty()) Q.pop();
for(int i = 0; i < N; i++){
int x;
scanf("%d",&x);
Q.push(x);
}
int ans = 0;
while(!Q.empty()&&Q.size()>1){
int a = Q.top();
Q.pop();
int b = Q.top();
Q.pop();
ans += a + b;
Q.push(a+b);
}
printf("%d\n",ans);
}
}
没有必要真的构造一个哈夫曼树,用小顶堆每次输出即可,构造一个虚拟的哈夫曼树。
来源:CSDN
作者:Leonjang
链接:https://blog.csdn.net/weixin_38023259/article/details/104117525