《算法笔记》9.8小节——图算法专题->哈夫曼树 问题 C: 哈夫曼树

                                            问题 C: 哈夫曼树

题目描述

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

输入

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

输出

输出权值。

样例输入

2
2 8 
3
5 11 30 

样例输出

10
62

AC代码：

#include<cstdio>
#include<queue>
using namespace std;
int main(){
    int n;
    long long temp,x,y;
    while(scanf("%d",&n)!=EOF){
        int ans = 0;
        priority_queue<long long ,vector<long long>,greater<long long> >q;
        for(int i = 0;i < n;i++){
          scanf("%lld",&temp);
          q.push(temp);
        }
        while(q.size() > 1){
          x = q.top();
          q.pop();
          y = q.top();
          q.pop();
          q.push(x+y);
          ans += x+y; 
        }
        q.pop();
        printf("%lld\n",ans);
    }
    return 0;
} 
/**************************************************************
    Problem: 1921
    User: 2015212040209
    Language: C++
    Result: 正确
    Time:0 ms
    Memory:1172 kb
****************************************************************/