【C++】STL使用仿函数控制优先级队列priority_queue

简介: 【C++】STL使用仿函数控制优先级队列priority_queue

前言

本文章讲解C++STL的容器适配器:priority_queue的实现,并实现仿函数控制priority_queue底层。


一、priority_queue的底层实现

priority_queue叫做优先级队列,它的底层结构是堆,在库中,默认生成的是大堆

在库的实现中,使用vector作为该优先级队列的适配容器。

由于priority_queue也是一个适配器,所以它的接口函数也可以对其他容器的函数进行封装使用。

下面来对priority_queue进行模拟实现。

#pragma once
//优先级队列底层是堆,heap
namespace bit
{
  //仿函数
  template<class T>
  class Less
  {
  public:
    bool operator()(const T& t1, const T& t2)
    {
      return t1 < t2;
    }
  };
  template<class T>
  class Greater
  {
  public:
    bool operator()(const T& t1, const T& t2)
    {
      return t1 > t2;
    }
  };
  //类名不是类型
  template<class T, class Container = vector<T>, class Compare = Less<T> >
  //默认大堆
  class PriorityQueue
  {
    //com是一个仿函数
    Compare com;
    void AdjustDown(int parent)
    {
      int child = parent * 2 + 1;
      while (child > 0)
      {
        //可能右孩子存在且大于左孩子
        if (child + 1 < _con.size() && com(_con[child],_con[child + 1]))
        {
          ++child;
        }
        //如果孩子存在且孩子大于父亲,交换
        if (child < _con.size() && com(_con[parent],_con[child]))
        {
          swap(_con[child], _con[parent]);
          parent = child;
          child = parent * 2 + 1;
        }
        else
        {
          break;
        }
      }
    }
    void AdjustUp(int child)
    {
      Compare com;
      //类名实例化类对象,该类型是一个仿函数,实例化的com可以调用仿函数的比较方法
      //记录父亲的下标
      int parent = (child - 1) / 2;
      while (child > 0)
      { 
        if (com(_con[parent], _con[child]))
        {
          swap(_con[child], _con[parent]);
        }
        child = parent;
        parent = (child - 1) / 2;
      }
    }
  public:
    PriorityQueue()
    {}
    //默认建大堆
    template<class InputIterator>
    PriorityQueue(InputIterator first, InputIterator end)
    {
      //插入数据
      while (first != end)
      {
        _con.push_back(*first);
        ++first;
      }
      //向下调整建堆
      for (int i = (_con.size() - 1 - 1) / 2; i >= 0; --i)
      {
        AdjustDown(i);
      }
    }
    void push(const T& val)
    {
      //插入元素
      _con.push_back(val);
      //然后向上调整
      AdjustUp(_con.size() - 1);
    }
    void pop()
    {
      //1.堆顶和堆尾交换
      swap(_con[0], _con[_con.size() - 1]);
      //2.删除堆尾
      _con.pop_back();
      //3.向下调整,恢复堆
      AdjustDown(0);
    }
    T& top() 
    {
       return _con[0];
    }
    size_t size() const
    {
      return _con.size();
    }
    bool empty() const
    {
      return _con.empty();
    }
  private:
    Container _con;

注解:

  • 1.在进行构造时,我们使用迭代区间进行构造
  • (1)向空间中插入数据
  • (2)向下调整建堆,建N个数据,从第一个非叶子节点开始进行建堆,每次都向下调整,时间复杂度O(N)。

push函数的实现:

向堆尾插入一个元素,插入的元素可能会改变堆的结构,所以我们需要将该元素向上调整,以维护堆的特性。

pop函数的实现:

删除堆顶元素,首先将堆顶元素和堆的最后一个元素进行交换,然后进行尾删,删除后原来的堆尾的元素现在在堆顶,我们需要将其进行向下调整以维持堆的状态。

empty函数:

判断堆是否为空即可。

size函数:

计算堆的元素个数。

top函数:

取堆顶元素,也就是取第一个元素。

priority_queue容器的作用:有需要使用堆的地方就可以使用该容器。

二、使用仿函数控制priority_queue的底层

什么是仿函数?

仿函数:仿造的函数,它并不是一个真正意义上的函数。能够重载operator(),内部实现自己想要的方法,然后实例化出一个方法对象,调用该对象作为参数,此时就可以调用该方法类对象内部实现的operator()方法了。

由于库中的优先级队列的底层结构是堆,且默认是大堆,我们在模拟实现和使用时,如果遇到需要使用小堆的场景,需要改变的东西很多,比如向上调整算法和向下调整算法的比较方法。

现在我们需要指定一个方法,这个方法可以由我们控制,从而实现大小堆。

仿函数:

//仿函数
template<class T>
class Less
{
public:
  bool operator()(const T& t1, const T& t2)
  {
    return t1 < t2;
  }
};
template<class T>
class Greater
{
public:
  bool operator()(const T& t1, const T& t2)
  {
    return t1 > t2;
  }
};

这里实现了两个类,并重载了一个比较方法。

在priority_queue模板的实现中,我们可以多传递一个模板参数:class Compare,也就是多传递一个比较方法。然后我们自己写一个比较方法传递过去,然后在priority_queue的实现中实例化一个方法对象来控制堆的生成。


总结

本文章讲解了仿函数控制priority_queue容器的底层实现的过程以及priority_queue的底层实现。

只要有堆的地方就可以使用priority_queue容器。

相关文章
|
2天前
|
存储 C++ 容器
C++:STL - set & map
C++:STL - set & map
11 4
|
3天前
|
算法 安全 程序员
【C++】STL学习之旅——初识STL,认识string类
现在我正式开始学习STL,这让我期待好久了,一想到不用手撕链表,手搓堆栈,心里非常爽
10 0
|
4天前
|
存储 Serverless C++
【C++入门到精通】哈希 (STL) _ unordered_map _ unordered_set [ C++入门 ]
【C++入门到精通】哈希 (STL) _ unordered_map _ unordered_set [ C++入门 ]
8 1
|
5天前
|
存储 设计模式 算法
【C++/STL】stack和queue(容器适配器、优先队列、双端队列)
【C++/STL】stack和queue(容器适配器、优先队列、双端队列)
12 1
|
5天前
|
存储 编译器 C++
【C++/STL】list(常见接口、模拟实现、反向迭代器、)
【C++/STL】list(常见接口、模拟实现、反向迭代器、)
5 0
|
4天前
|
设计模式 安全 算法
【C++入门到精通】特殊类的设计 | 单例模式 [ C++入门 ]
【C++入门到精通】特殊类的设计 | 单例模式 [ C++入门 ]
15 0
|
6天前
|
C语言 C++
【C++】string类(常用接口)
【C++】string类(常用接口)
15 1
|
3天前
|
编译器 C++
【C++】继续学习 string类 吧
首先不得不说的是由于历史原因,string的接口多达130多个,简直冗杂… 所以学习过程中,我们只需要选取常用的,好用的来进行使用即可(有种垃圾堆里翻美食的感觉)
7 1
|
3天前
|
存储 安全 测试技术
【C++】string学习 — 手搓string类项目
C++ 的 string 类是 C++ 标准库中提供的一个用于处理字符串的类。它在 C++ 的历史中扮演了重要的角色,为字符串处理提供了更加方便、高效的方法。
13 0
【C++】string学习 — 手搓string类项目
|
4天前
|
Java C++ Python
【C++从练气到飞升】06---重识类和对象(二)
【C++从练气到飞升】06---重识类和对象(二)