今天才知道,哈夫曼树的权值就等于所有非叶子节点的和。
求哈夫曼树要用小顶堆。
代码实现:
#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=1;i<=n;i++){
int x;
scanf("%d",&x);
Q.push(x);
}
int ans=0;
while(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);
}
return 0;
}