poj 3253 Fence Repair(模拟huffman树 + 优先队列)

题意：如果要切断一个长度为a的木条需要花费代价a， 问要切出要求的n个木条所需的最小代价。

思路：模拟huffman树，每次选取最小的两个数加入结果，再将这两个数的和加入队列。

注意priority_queue的用法，原型：

priority_queue<Type> q;  
priority_queue<Type,deque<Type>,Comp> q;  

其中Type是类型，Comp是比较结构体，比较函数是它的括号重载，比如对int型从小到大排序的Comp结构体如下：

struct Comp  
{  
    bool operator()(const LL& a,const LL&b) const  
    {  
        return a>b;  
    }  
};  

还有要注意的是，不能直接while(!q.empty())，因为每次选取两个数，如果队列只剩下一个数时又要取两个数，此时程序会爆掉，应该用一个while(1)循环，在取出一个数时立刻判断if(q.empty()) break;

不能直接模拟huffman树，直接huffman会超时。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<math.h>
#include<algorithm>
#include<queue>
#include<set>
#include<bitset>
#include<map>
#include<vector>
#include<stdlib.h>
#include <stack>
using namespace std;
#define PI acos(-1.0)
#define max(a,b) (a) > (b) ? (a) : (b)
#define min(a,b) (a) < (b) ? (a) : (b)
#define ll long long
#define eps 1e-10
#define MOD 1000000007
#define N 1000000
#define inf 1e12
ll n;
struct Comp{
   bool operator()(const ll &a,const ll &b)const{
      return a>b;
   }
};
int main()
{
   while(scanf("%I64d",&n)==1){
      priority_queue<ll,deque<ll>,Comp> q;
      for(int i=1;i<=n;i++){
         ll x;
         scanf("%I64d",&x);
         q.push(x);
      }
      ll ans=0;
      while(1){
         ll L=q.top(); q.pop();
         if(q.empty()) break;
         ll R=q.top(); q.pop();
         ans+=L;
         ans+=R;
         q.push(L+R);
      }
      printf("%I64d
",ans);
   }
    return 0;
}