2.Huffman树-哈夫曼树
实现思路:构建一颗哈夫曼树,求最短带权路径长度(树中所有的叶结点的权值乘上其到根结点的路径长度)
- 每次选择重量最小的两堆进行合并
- 使用小根堆存储每一堆果子,每次两次弹出堆顶元素,合并后就放入小根堆中
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
int main(){
int n;
cin>>n;
priority_queue<int ,vector<int>,greater<int>> heap;
while(n--){
int x;
cin>>x;
heap.push(x);
}
int res=0;
while(heap.size()>1){
int a=heap.top(),heap.pop();
int b=heap.top(),heap.pop();
res+=a+b;
heap.push(a+b);
}
cout<<res<<endl;
return 0;
}