优先级队列

本文介绍了C++标准库中的优先级队列priority_queue如何使用,特别是如何通过传入仿函数greater<>或less<>来创建小堆和大堆。文中详细讲解了堆的向上和向下调整过程,并展示了如何自定义仿函数以适应不同类型的数据比较,如Date类型和指针类型。此外,还提供了push、pop、top等操作的实现示例。
摘要由CSDN通过智能技术生成

优先级队列priority_queue

				//需要建立小堆则需传入三个参数,第三个判断小堆(升序)
				priority_queue<int,vector<int>,greater<int>> pq1;
        pq1.push(1);
        pq1.push(3);
        pq1.push(9);
        pq1.push(4);
        pq1.push(7);
        pq1.push(0);
        pq1.push(2);
        while(!pq1.empty())
        {
            cout << pq1.top() << " ";
            pq1.pop();
        }
        cout << endl;

优先级队列就是以堆为基础的队列,当插入一个值时,自动建堆,默认为大堆less<>如果需要建立小堆,需要传入三个参数,默认第二个为vector类型,greater<>为建立小堆的仿函数。

每次top中的值总是极值,用堆来实现队列中数据的排序。

priority_queue的实现

class priority_queue
    {
    public:
        //向上调整:尾部插入新值,找到合适位置
        void adjust_up(int child)
        {
            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 parent)
        {
            size_t child = parent * 2 + 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 = parent * 2 + 1;
                }
                else
                {
                    break;
                }
            }
        }
		private:
        Container _con;
}

优先级队列就是对堆的操作,所以向上和向下两个调整是必须的,每当执行pop和push时,数据变化都要调整堆:父节点大于子节点时交换,那么建立的就是小堆,父节点小于子节点时交换,那么建立的就是小堆,这是堆的调整,也是建堆的基础。

但是我们会发现,当我们在定义队列时,无法做到修改类中函数的大于小于号,所以可以建立仿函数,来让用户自定义是大堆还是小堆。

仿函数

//仿函数
    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;
        }
    };

以前我们对堆调整时,需要判断数据的大小:

父节点大于子节点时交换,那么建立的就是小堆,父节点小于子节点时交换,那么建立的就是小堆,这里的大于小于号以前是需要手动变的,现在有了仿函数,只需要在实例化的时候传入greater或less参数,就能自动识别是大堆的建立还是小堆的建立。

				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();
        }

基本的操作实现。

针对类型构建仿函数:

void test2()
    {
        // 大堆,需要用户在自定义类型中提供<的重载
        priority_queue<Date, vector<Date>, greater<Date>> q1;
        q1.push(Date(2018, 10, 29));
        q1.push(Date(2018, 10, 30));
        q1.push(Date(2018, 10, 28));
        cout << q1.top() << endl;

        priority_queue<Date*, vector<Date*>, PDateGreater> q2;
        q2.push(new Date(2018, 10, 29));//传的是地址,要新建仿函数
        q2.push(new Date(2018, 10, 30));
        q2.push(new Date(2018, 10, 28));
        cout << *(q2.top()) << endl;
    }

我们可以发现,q1传入的是Date类型的值,可以直接使用上面我们建立的仿函数;在q2中,当我们传入的是new出来的地址,那么就需要建立新的仿函数来传地址类型的,解引用后再比较大小。

class PDateLess
    {
    public:
        bool operator()(const Date* p1, const Date* p2)
        {
            return *p1 < *p2;
        }
    };
    class PDateGreater
    {
    public:
        bool operator()(const Date* p1, const Date* p2)
        {
            return *p1 > *p2;
        }
    };
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值