题目1172:哈夫曼树 九度

题目描述:

哈夫曼树,第一行输入一个数n,表示叶结点的个数。需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即weight,题目需要输出所有结点的值与权值的乘积之和。

输入:

输入有多组数据。
每组第一行输入一个数n,接着输入n个叶节点(叶节点权值不超过100,2<=n<=1000)。

输出:

输出权值。

样例输入:
5  
1 2 2 5 9
样例输出:
37

代码:
#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()==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;    
} 


你可能感兴趣的:(哈夫曼树)