POJ3253哈弗曼树的优先队列解法

直接构造优先队列，每次取出最小的两个数相加，直到队列中只有一个数为止，还是用STL过的题。

 

注意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;}};

 

这题还要注意使用long long，不然会越界导致WA。

 

我的代码：

#include <iostream>#include <cstdio>#include <cstdlib>#include <cstring>#include <queue>using namespace std;typedef long long LL;struct Comp{bool operator()(const LL& a,const LL&b) const{return a>b;}};priority_queue<LL,deque<LL>,Comp> q;void init(){LL n,a;while(!q.empty())q.pop();scanf("%lld",&n);while(n--){scanf("%lld",&a);q.push(a);}}LL run(){LL ret=0;LL a,b;while(1){a=q.top();q.pop();if(q.empty())break;b=q.top();q.pop();a+=b;ret+=a;q.push(a);}return ret;}int main(){init();printf("%lld/n",run());return 0;}

 

总结：做题的时候要注意数字的范围，看清到底要用什么类型，是int还是long long。