优先级队列

#pragma once
namespace abc
{
    template <class T, class Container = vector<T>, class Compare = std::less<T> >
    class priority_queue
    {
    public:
        priority_queue()
        {}
        template <class InputIterator>
        priority_queue(InputIterator first, InputIterator last)
            :c(first, last)
        {
            for (int i = (c.size() - 2) / 2; i >= 0; --i)
            {
                ad_just_down(i);
            }
        }

        bool empty() const
        {
            return c.empty();
        }

        size_t size() const
        {
            return c.size();
        }

        const T& top() const
        {
            return c[0];
        }
        void ad_just_down(int pos)
        {
            int parent = pos;
            int child = parent * 2 + 1;
            //现在孩子是最大的那个
            while (child < c.size())//直到没有孩子结束
            {
                if (child < (int)(c.size() - 1) && Compare()(c[child], c[child + 1]))
                {
                    child++;//找最大孩子
                }
                std::swap(c[parent], c[child]);
                parent = child;//更新孩子和父亲
                child = parent * 2 + 1;
            }
        }
        void ad_just_up(int pos)
        {
            int child = pos;
            int parent = (child - 1) / 2;
            while (child > 0)//只要父亲存在就继续
            {
                if (Compare{}(c[parent], c[child]))//父亲小于孩子
                {
                    std::swap(c[parent], c[child]);
                    child = parent;
                    parent = (child - 1) / 2;
                }
                else
                    break;
            }
        }
        void push(const T& x)
        {
            c.push_back(x);
            ad_just_up(c.size() - 1);
        }

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

    private:
        Container c;
        Compare comp;
    };
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值