用标准模板库定义小顶堆,求哈夫曼树带权和
要求:输入一个整数n,在输入n个整数,输出带权和
输入样例:5
1 2 2 5 9
输出样例:
37
实现代码:
#include <iostream>
#include <queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> > Q;
int main(int argc, char *argv[])
{
int n;
while(scanf("%d",&n)!=EOF){
while(Q.empty()==false) 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;
}
测试样例: