priority_queue用法

from: http://blog.csdn.net/silentskydream/archive/2009/04/14/4073111.aspx

 

今天在写堆和哈夫曼树的ACM题的时候,接触到priority_queue的用法,由于比较函数的难写,请教过队内的红薯和杨大牛后才稍微弄明白些,下面总结如下,首先我是用手写的堆来过题的,其实和照黑书指导上的那个堆的代码差不多。

写完之后就看了下STL里面的priority_queue的用法就开始研究,首先是用了网上找的一个写比较函数的方法是用操作符重载做的。代码如下:

//比较函数
对于结构体
struct heapmin
{
 heapmin(int tx){x=tx;};
 int x;
};
struct heapmax
{
 heapmax(int tx){x=tx;};
 int x;
};
bool operator<(const struct heapmin &a,const struct heapmin &b)
{
  return a.x<b.x;
}//最大堆
bool operator<(const struct heapmax &a,const struct heapmax &b)
{
  return a.x>b.x;
}//最小堆

然后就可以用STL里面给的那些push,pop,top,size,empty函数了。

然后由于G++一直跑的RE所以我想把结构体改成里面只是存int型整数,然后就瓜起了,不会写了,自己仿照sort里面那个比较函数写是错的,然后我就不会写了,就去群里问人,得到两种方法,

一个是用stl里面#include<functional>里面的great<int> less<int>最比较函数写,我试了下,是可以的。谢谢红薯;

代码如下:

//比较函数
#inclulde<iostream>
#include<functional>
priority_queue< int, vector<int>, great<int> >//最小堆
priority_queue< int, vector<int>, less<int> >//最大堆

然后就是杨大牛的方法,把比较函数写成结构体的形式,代码如下:

#include<iostream>
#include<queue>
using namespace std;
struct cmp
{
 bool operator()(const int &a,const int &b)
 {
  return a>b;//最小堆
  return a<b;//最大堆
 }
};
priority_queue< int, vector<int>, cmp >

用了上面的所有方法做了poj1442

http://162.105.81.212/JudgeOnline/problem?id=1442

你可能感兴趣的:(struct)