九度OJ 1172

题目描述：

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

输入：

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

输出：

输出权值。

样例输入：

5  
1 2 2 5 9

样例输出：

37

题目中已经说明这是一个哈夫曼树的例子，一般在OJ中出现的都是这种求最小带权路径长度和。在给定的n个点中，我们每次选择最小的两个点，将它们求和之后在放入集合中，重复操作，直到集合中剩下一个元素，也就是哈夫曼树的根节点。我们将所有的非叶子节点相加，结果就是最小带权路径长度和。在这里，我们可以利用堆这个数据结构求的集合中最小的元素。因为库函数中默认的是大顶堆，也就是堆顶元素为最大的元素，我们只要定义一个小顶堆即可。  priority_queue<int,vector<int>,greater<int> >Q;  

这里千万要注意的一点是 最后两个>>之间要有空格，> >,因为如果没有空格的话，低级的编译器或者OJ上默认是把它当成cin>>的>>,为了避免歧义，我们还是加上空格为好。

#include <iostream>
#include <queue>
using namespace std;

int main(){
    int n;
    int a[1000];
    priority_queue<int, vector<int>, greater<int> > Q;
    while(cin>>n){
        for(int i=0;i<n;i++){
            cin>>a[i];
            Q.push(a[i]);
        }
        int result = 0;
        while(!Q.empty()){
            int a = Q.top();
            Q.pop();
            int b = 0;
            if(!Q.empty()){
                b = Q.top();
                Q.pop();
            }
            else{
                break;
            }
            result = result + a + b;
            Q.push(a+b);
        }
        cout<<result<<endl;
    }
    return 0;
}