九度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;}