C++:priority_queue模拟实现

简介: C++:priority_queue模拟实现



什么是priority_queue

priority_queue称为优先级队列。优先级队列是一种特殊的队列,其中每个元素都有一个相关的优先级。元素的优先级决定了它们在队列中的顺序。具有较高优先级的元素在队列中被排在前面,而具有较低优先级的元素在队列中被排在后面。

可以简单理解为,priority_queue其实就是堆,如果作用于整型,可以让最大/最小的数字处于堆顶,而其也可以作用于其它类型,比如作用于string,那么可以根据字典序来排序。

接下来我带大家实现一下这个结构:


模拟实现

在STL中,preority_queue是一个容器适配器,因为其只要求了数据按照指定的顺序排列,但是没有对底层结构做要求,所以我们直接用其他容器做底层即可。

在此我使用了vector做底层,因为在优先级队列中,我们需要频繁地进行下标访问,使用deque也是可以的。

基本结构:

template <class T, class Container = vector<T>>
class preority_queue
{
private:
  Container _con;
};

模板参数:

T:这个优先级队列承载的元素类型

Container :底层容器类型,此处默认值为vector

我们看看preority_queue有哪些接口:

接下来我们一一实现:


top:

想要得到preority_queue优先级最高的元素,其实就是拿到vector的第一个元素_con[0]

const T& top()
{
  return _con[0];
}

size:

preority_queue的大小,就是vector的大小。

size_t size()
{
  return _con.size();
}

empty:

判断preority_queue是否为空,就是判断vector是否为空。

bool empty()
{
  return _con.empty();
}

向上调整算法

现在假设我有以下堆结构:

我现在在堆尾部插入一个数据,如何将数据调整到合适的位置,保证这个结构依然满足堆的要求?

想要将其插入到指定位置,就要使用到向上调整算法:将最后一个节点向上调整到合适的位置

首先讲解一个公式:堆结构中父节点与子节点的下标关系

假设父节点的下标为parent,则其左子节点的下标为 2 * parent+1,右子节点的下标为 2 * parent+2。

由于大堆要保证每隔父亲节点大于两个子节点,而除去最后一个节点,其它的节点已经满足堆结构了,所以此处需要将最后一个节点不断地与其父亲节点比较,如果其比父亲节点大,就交换位置,然后继续和新的父亲节点比较,直到比当前的父亲节点小,或者到达堆顶为止。

图示:

vector中的效果(实际效果):

代码实现:

void AdjustUp(int child)
{
  int parent = (child - 1) / 2;
  while (child > 0)
  {
    if (_con[child] > _con[parent])
    {
      Swap(_con[child], _con[parent]);
      child = parent;
      parent = (child - 1) / 2;
    }
    else
    {
      break;
    }
  }
}

代码详解:

  1. 定义父节点索引:
int parent = (child - 1) / 2;

根据完全二叉树的性质,节点i的父节点索引是(i - 1) / 2,所以计算出child节点的父节点索引。

  1. 进入循环:
while (child > 0)

child大于0时,继续执行循环。循环的目的是将child节点与其父节点进行比较,并根据需要进行交换。

  1. 比较child节点与其父节点的大小:
if (_con[child] < _con[parent])

如果child节点的值小于其父节点的值,说明需要进行交换。这是一个小堆的向上调整操作。如果想要实现大堆的向上调整,需要将判断条件改为>

  1. 交换节点值和更新索引:
Swap(_con[child], _con[parent]);
child = parent;
parent = (child - 1) / 2;

交换child节点和父节点的值,然后更新childparent的值,使其指向交换后的节点。

  1. 循环结束:
else
{
    break;
}

如果child节点大于等于其父节点,说明调整完成,跳出循环。

通过这个向上调整的操作,可以将新插入的元素调整到合适的位置,以保证堆的性质。


向下调整算法

如果堆中某个节点的值被修改,如何调整这个堆的结构保证其依然满足堆的要求?

当堆中的某个节点的值发生改变时(例如,该节点的值被修改),需要进行向下调整操作来保持堆的性质。

向下调整的基本思想是将当前节点与其子节点进行比较,并根据堆的类型(大堆或小堆)选择合适的交换操作。如果当前节点的值小于(或大于)其子节点的值,那么需要将当前节点与其子节点中的较大(或较小)值进行交换。然后,继续向下调整交换后的子节点,直到满足堆的性质为止。

示意图如下:

vector中的效果(实际效果):

代码如下:

void AdjustDown(int parent)
{
  int child = parent * 2 + 1;
  while (child < _con.size())
  {
    if (child + 1 < con.size() && _con[child + 1] > _con[child])
    {
      child++;
    }
    if (_con[child] > _con[parent])
    {
      Swap(_con[child], _con[parent]);
      parent = child;
      child = parent * 2 + 1;
    }
    else
    {
      break;
    }
  }
}

首先,计算待调整节点的左子节点下标child = parent * 2 + 1

然后,进入一个循环,判断child是否越界。如果child < size,则说明待调整节点有左子节点。

在循环中,首先判断是否存在右子节点,并且右子节点的值大于左子节点的值,如果满足这个条件,则将child更新为右子节点的下标。

接下来,判断child节点的值是否大于parent节点的值。如果满足这个条件,则交换childparent节点的值,并更新parentchild,再次计算child的值。

如果child节点的值不大于parent节点的值,则退出循环。

函数结束后,即可保证以parent节点为根的子树满足堆的性质。


插入与删除

push:

我们只需要将待插入元素放到末尾,然后将其向上调整即可:

void push(const T& x)
{
  _con.push_back(x);
  adjust_up(_con.size() - 1);
}

pop:

我们只需要将第一个元素与最后一个元素交换,然后删除最后一个元素,最后把刚刚交换上来的第一个元素向下调整即可:

void pop()
{
  swap(_con[0], _con[_con.size() - 1]);
  _con.pop_back();
  adjust_down(0);
}

但是至此我们的优先级队列还不是完全体,因为其只能固定是大堆/小堆,或者说优先级的比较方式是固定的,想要解决这个问题,我们就需要仿函数。


仿函数

仿函数本质上是一个类,它具有函数调用运算符(operator())的特性。仿函数的本质是一种可调用的对象,它可以像函数一样被调用。

我先为大家展示一个仿函数:

class Add 
{
    int operator()(int a, int b) 
    {
        return a + b;
    }
};
int main()
 {
    Add add;
    int result = add(3, 4); 
    
    return 0;
}

在上面的示例中,我们定义了一个名为Add的类,它重载了函数调用运算符operator(),并实现了对两个整数相加的逻辑。在主函数中,我们创建了一个Add的实例add,并通过调用函数调用运算符对两个整数进行相加。

add本质上明明是一个类的对象,但是由于重载了运算符operator(),所以可以模仿函数的行为,完成函数的调用。

那么我们为什么不直接写一个函数,完成两个数字的加法,而是要搞仿函数这样的东西呢?

这就不得不提到被人诟病的C语言函数指针体系了,在C语言中,如果我们想将一个函数作为参数传给其他函数,那么我们就要在形参列表中准确写出函数指针的类型。但是函数指针经常会非常复杂,用起来很难受。

但是对象就不一样了,如果我们将仿函数的对象作为参数传递,此时的类型就非常好写了。

其次是内联函数的问题:

仿函数是定义在类中的,类中的成员函数默认为内联函数,此时短小的函数就会被展开。而如果是一般的函数,用inline修饰后,由于要在指定位置展开,此时就没有地址,没有函数指针了,所以内联函数无法作为参数传递。如果inline修饰的函数被作为函数指针传递,此时这个函数的inline就会失效,从而无法展开。而仿函数很好解决了这个问题

所以仿函数有两大优点:

  1. 避免了复杂的函数指针体系,我们可以很方便的将仿函数作为参数传递
  2. 仿函数可以作为参数传递的同时,还支持内联函数

对于我们的优先级队列,我们就可以使用仿函数作为一个模板参数。

用户使用优先级队列时,可以自己传入一个仿函数,从而制定自己的规则,按照自己设想的方法来排列数据。

比大小的仿函数:

template <class T>
struct less
{
  bool operator()(const T& x, const T& y)
  {
    return x < y;
  }
};
template <class T>
struct greater
{
  bool operator()(const T& x, const T& y)
  {
    return x > y;
  }
};

完整版priority_queue

template <class T, class Container = vector<T>, class Compare = less<T>>
class priority_queue
{
public:
  void adjust_up(int x)
  {
    int child = x;
    int parent = (child - 1) / 2;
    while (child > 0)
    {
      if (Compare()(_con[parent], _con[child]))
      {
        swap(_con[child], _con[parent]);
        child = parent;
        parent = (child - 1) / 2;
      }
      else
      {
        break;
      }
    }
  }
  void adjust_down(int x)
  {
    int parent = x;
    int child = 2 * parent + 1;
    while (child < _con.size())
    {
      if (child + 1 < _con.size() && Compare()(_con[child], _con[child + 1]))
      {
        child++;
      }
      if (Compare()(_con[parent], _con[child]))
      {
        swap(_con[child], _con[parent]);
        parent = child;
        child = 2 * parent + 1;
      }
      else
      {
        break;
      }
    }
  }
  void push(const T& x)
  {
    _con.push_back(x);
    adjust_up(_con.size() - 1);
  }
  void pop()
  {
    swap(_con[0], _con[_con.size() - 1]);
    _con.pop_back();
    adjust_down(0);
  }
  const T& top()
  {
    return _con[0];
  }
  size_t size()
  {
    return _con.size();
  }
  bool empty()
  {
    return _con.empty();
  }
private:
  Container _con;
};

在模板参数中,我们传入了第二个参数,用于接受一个仿函数Compare ,这个仿函数用于决定后续比较时的优先级规则。

后续我们在比较parentchild大小的地方调用了仿函数:Compare()(_con[parent], _con[child])

这里并没有多写一对括号,Compare()是一个匿名对象,而后面的(_con[parent], _con[child])则是在调用operator()

当然,如果你认为这样可读性太差了,也可以在类中多定义一个compare类的对象cmp,后续直接cmp(_con[parent], _con[child])这样调用这个仿函数。


相关文章
|
19小时前
|
存储 算法 C++
c++的学习之路:17、stack、queue与priority_queue
c++的学习之路:17、stack、queue与priority_queue
31 0
|
19小时前
|
存储 算法 C语言
【C++入门到精通】C++入门 —— priority_queue(STL)优先队列
本文介绍了C++的STL组件`std::priority_queue`,它是一个容器适配器,实现优先队列数据结构,通常基于堆实现。文章讨论了优先队列的基本概念、特点和底层堆结构,强调了其自动排序和优先级最高元素的访问。还展示了如何定义、插入、访问和移除元素,以及自定义比较函数。此外,提供了模拟实现`priority_queue`的代码示例,探讨了仿函数的作用,包括默认的`std::less`和自定义比较函数。文章鼓励读者进一步探索C++的优先队列及其应用。
27 3
|
19小时前
|
存储 算法 C++
【C++初阶】STL详解(九) priority_queue的使用与模拟实现
【C++初阶】STL详解(九) priority_queue的使用与模拟实现
21 0
|
19小时前
|
存储 算法 C++
C++:stack、queue、priority_queue增删查改模拟实现、deque底层原理
C++:stack、queue、priority_queue增删查改模拟实现、deque底层原理
35 0
|
19小时前
|
算法 C++ 容器
【C++练级之路】【Lv.10】【STL】priority_queue类和反向迭代器的模拟实现
【C++练级之路】【Lv.10】【STL】priority_queue类和反向迭代器的模拟实现
|
19小时前
|
存储 算法 调度
【C/C++ 数据结构 优先队列】了解学习`std::priority_queue`的使用
【C/C++ 数据结构 优先队列】了解学习`std::priority_queue`的使用
48 3
|
19小时前
|
算法 安全 编译器
C++:模版进阶 | Priority_queue的模拟实现
C++:模版进阶 | Priority_queue的模拟实现
|
7月前
|
算法 C++ 容器
C++初阶之一篇文章教会你queue和priority_queue(理解使用和模拟实现)(下)
优先队列是一种容器适配器,根据严格的弱排序标准,它的第一个元素总是它所包含的元素中最大的。 此上下文类似于堆,在堆中可以随时插入元素,并且只能检索最大堆元素(优先队列中位于顶部的元素)。 优先队列被实现为容器适配器,容器适配器即将特定容器类封装作为其底层容器
|
9月前
|
存储 算法 C++
C++ STL priority_queue
优先队列是一种容器适配器,根据严格的弱排序标准,它的第一个元素总是它所包含的元素中最大的。
|
19小时前
|
测试技术 C++
c++优先队列priority_queue(自定义比较函数)
c++优先队列priority_queue(自定义比较函数)
82 0