九度1172解题报告

http://ac.jobdu.com/problem.php?pid=1172

这道题原来做过可是一点印象都没有，竟然还是能想到用priority_queue来做，看来之前的练习还是有效果。其实今年考研408的算法题与这道题类似，但我当时并不知道这个结论：哈夫曼树所有叶节点的权重之和等于所有非叶结点的值的和，所以当时费了大量的时间做那道题，以至于组成原理的两道大题几乎完全没做，想想就心痛。不过如今逆袭了，也算比较心安。因为构造哈夫曼树的时候总是要取两个权值最小的结点，构造好的新非叶结点又要重新放回去筛选，如果反复排序会大大增加时间复杂度，所以用小顶堆解决这个问题就会容易许多，即用STL里的priority_queue来实现。代码如下：

#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()) q.pop();
       for (int i=0;i<n;++i)
       {
           int x;
           scanf("%d",&x);
           q.push(x);
       }
       if (n==1)
       {
           printf("%d\n",q.top());
           continue;     
       }
       int ans=0;
       while(q.size()>1)
       {
           int temp=q.top();
           q.pop();
           temp+=q.top();
           q.pop();
           ans+=temp;
           q.push(temp);  
       } 
       printf("%d\n",ans);
   }
   return 0;
}