优先级队列默认使用vector作为其底层存储数据的容器,在vector上又使用了堆算法将vector中元素构造成堆的结构,因此priority_queue就是堆,所有需要用到堆的位置,都可以考虑使用priority_queue。注意:默认情况下priority_queue是大堆。
归根结底:
我们就是用vector作为适配器实现一个大堆。
使用堆算法就是进行parent和child的比较,不满足条件就交换parent和child。
默认的堆是一个大堆,但是我今天设计的是一个默认小堆的函数,而且我的lesser是小堆比较,这是因为和单词进行对比更容易上手理解。(记住与实际相反就好了,实际中less是大堆,greater是小堆)
#pragma once
#include
#include
#include
using namespace std;
namespace tom
{
template <class T, class Container = vector<T>, class Compare = lesser<T> >
class priority_queue
{
public:
priority_queue()
:c()
{
}
template <class InputIterator>
priority_queue(InputIterator first, InputIterator last)
:c(first,last)
{
int sz = c.size();
int root = (sz-1)/ 2;//parent=(child-1)/2 ; child=(parent+1)*2;
while (root >= 0)
{
AdjustDown(root);
root--;
}
}
bool empty() const
{
return c.empty();
}
size_t size() const
{
return c.size();
}
const T& top() const
{
if (!c.empty())
return c[0];//c.front();
else
cout << "空堆,请插入元素!" << endl;
}
void push(const T& x)
{
c.push_back(x);
AdjustUP(c.size() - 1);
}
void pop()//堆一般都是pop头的,pop尾没啥意思
{
if (!empty())
{
swap(c.front(), c.back());//swap(c[0], c[c.size()-1]);
c.pop_back();
AdjustDown(0);
}
}
private:
Container c;
Compare comp;
};
};
先构建一个命名空间这是一个良好的习惯
设计一个类priority_queue剩下的就交给vector了,因为stl会把你给的数据直接初始化的;
但是要进行堆的排序还是需要向下调整的(AdjustDown)于是乎要设计一个函数AdjustDown;
并且删除数据或者插入数据都需要调整,于是乎在设计一个函数AdjustUp进行向上调整;
适配器
template <class T, class Container = vector<T>, class Compare = lesser<T> >
class priority_queue
{
...
}
这里的 class Container 就是一个适配器模板,c++不仅给了参数模板,还给了适配器模板,太逆天了!!!
这个适配器模板可以根据你的需求进行更改,比如改为deque,但是我们给了缺省值(默认就用vector适配器了)。
需要设计一个迭代器赋值构造其实是复用了vector的构造函数。
// 向上调整
void AdjustUP(int child)
{
int parent = ((child - 1) >> 1);
while (child)
{
if (c[child]<c[parent])
{
swap(c[child], c[parent]);
child = parent;
parent = ((child - 1) >> 1);
}
else
return;
}
}
// 向下调整
void AdjustDown(int parent)
{
int child = (parent)*2+1;
//parent = child / 2 - 1;
while (child < c.size())
{
if ((child + 1) < c.size() &&c[child + 1]<c[child])
{
child++;
}
if (c[child]<c[parent])
{
swap(c[child], c[parent]);
parent = child;
child = (parent) * 2 + 1;
}
else
return;
}
}
并不想再写一个大堆出来,因为会有代码冗余,于是乎想到了构造一个仿函数,来帮我实现大于小于的函数判断。
template<class T>
struct lesser
{
bool operator()(const T& x, const T& y)
{
return x < y;//我设计的与stl库设计的不一样,我把less设计成了小堆
}
};
template<class T>
struct greater
{
bool operator()(const T& x, const T& y)
{
return x > y;//我设计的与stl库设计的不一样,我把great设计成了大堆
}
};
template <class T, class Container = vector<T>, class Compare = lesser<T> >
class priority_queue
{
...
}
而且我在前面也留了扣子,在实现类模板的时候已经预留了compare函数了。
template <class T, class Container = vector<T>, class Compare = lesser<T> >
class priority_queue
{
...
}
其他细节
其他的细节基本没什么重点了。
#pragma once
#include
#include
#include
using namespace std;
namespace tom
{
template<class T>
struct lesser
{
bool operator()(const T& x, const T& y)
{
return x < y;//我设计的与stl库设计的不一样,我把less设计成了小堆
}
};
template<class T>
struct greater
{
bool operator()(const T& x, const T& y)
{
return x > y;//我设计的与stl库设计的不一样,我把great设计成了大堆
}
};
template <class T, class Container = vector<T>, class Compare = lesser<T> >
class priority_queue
{
public:
priority_queue()
:c()
{
}
template <class InputIterator>
priority_queue(InputIterator first, InputIterator last)
:c(first,last)
{
int sz = c.size();
int root = (sz-1)/ 2;//parent=(child-1)/2 ; child=(parent+1)*2;
while (root >= 0)
{
AdjustDown(root);
root--;
}
}
bool empty() const
{
return c.empty();
}
size_t size() const
{
return c.size();
}
const T& top() const
{
if (!c.empty())
return c[0];//c.front();
else
cout << "空堆,请插入元素!" << endl;
}
void push(const T& x)
{
c.push_back(x);
AdjustUP(c.size() - 1);
}
void pop()//堆一般都是pop头的,pop尾没啥意思
{
if (!empty())
{
swap(c.front(), c.back());//swap(c[0], c[c.size()-1]);
c.pop_back();
AdjustDown(0);
}
}
// 向上调整
void AdjustUP(int child)
{
int parent = ((child - 1) >> 1);
while (child)
{
if (Compare()(c[child] , c[parent]))
//if (c[child]
{
swap(c[child], c[parent]);
child = parent;
parent = ((child - 1) >> 1);
}
else
return;
}
}
// 向下调整
void AdjustDown(int parent)
{
int child = (parent)*2+1;
//parent = child / 2 - 1;
while (child < c.size())
{
if ((child + 1) < c.size() &&Compare()( c[child + 1],c[child]))
//if ((child + 1) < c.size() &&c[child + 1]
{
child++;
}
if (Compare()(c[child] , c[parent]))
//if (c[child]
{
swap(c[child], c[parent]);
parent = child;
child = (parent) * 2 + 1;
}
else
return;
}
}
private:
Container c;
Compare comp;
};
};