九度 题目1172：哈夫曼树

题目来源：http://ac.jobdu.com/problem.php?pid=1172

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：4471

解决：1889

题目描述：

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

输入：

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

输出：

输出权值。

样例输入：

5  
1 2 2 5 9

样例输出：

37

来源：

2010年北京邮电大学计算机研究生机试真题

利用优先级队列比较省事。

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>

using namespace std;

int main()
{
    int iData, iSum, n,i;
    priority_queue <int, vector<int>, greater <int> > Q;
    while(~scanf("%d", &n))
    {
        for(i = 0; i < n; ++i)
        {
            scanf("%d", &iData);
            Q.push(iData);
        }
        iSum = 0, i = 1;
        while(!Q.empty())
        {
            int a = Q.top();
            Q.pop();
            if(!Q.empty())
            {
                int b = Q.top();
                Q.pop();
                iSum += a+b;
                Q.push(a+b);
            }
        }
        printf("%d\n", iSum);
    }
    return 0;
}