算法 & 数据结构——优先队列

具备堆特性的队列,最大或最小的节点永远在第一个。

template <class T>
class PriorityQueue {
public:
    using Iter = typename std::vector<T>::iterator;

public:
    void Init()
    {
        for (auto i = _c.size() / 2; i != 0;)
        {
            Init(_c.begin(), --i, _c.size());
        }
    }

    void Push(const T & t)
    {
        _c.push_back(t);
        auto val = std::move(_c.back());
        auto idx = _c.end() - _c.begin();
        Push(_c.begin(), --idx, std::move(val));
    }

    void Pop()
    {
        auto len = _c.end() - _c.begin();
        std::swap(_c.front(), _c.back());
        Init(_c.begin(), 0, len - 1);
        _c.pop_back();
    }

    T & Top()
    {
        return _c.front();
    }

    bool Empty()
    {
        return _c.empty();
    }

    size_t Length()
    {
        return _c.size();
    }

    const std::vector<T> & Get()
    {
        return _c;
    }

    void Sort()
    {
        for (auto i = _c.size(); i != 0;)
        {
            --i;
            std::swap(_c.at(0), _c.at(i));
            Init(_c.begin(), 0, i);
        }
    }

private:
    void Init(Iter first, size_t idx, size_t len)
    {
        if (idx < len / 2)
        {
            auto max = idx;
            auto l = idx * 2 + 1;
            auto r = idx * 2 + 2;
            if (l < len && *(first + max) < *(first + l))
            {
                max = l;
            }
            if (r < len && *(first + max) < *(first + r))
            {
                max = r;
            }
            if (max != idx)
            {
                std::swap(  *(first + idx), 
                            *(first + max));
                Init(first, max, len);
            }
        }
    }
    
    void Push(Iter first, size_t idx, T && val)
    {
        for (auto   par = (idx - 1) / 2;
                    idx > 0 && *(first + par) < val;
                    par = (idx - 1) / 2)
        {
            *(first + idx) = std::move(*(first + par));
            idx = par;
        }
        *(first + idx) = std::move(val);
    }
    
private:
    std::vector<T> _c;
};

Push被调用时,新增的节点会通过<运算符放置在合适的位置
Pop被调用时,弹出队列元素,并且调整堆。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 1 序 2016年6月25日夜,帝都,天下着大雨,拖着行李箱和同学在校门口照了最后一张合照,搬离寝室打车去了提前租...
    RichardJieChen阅读 5,225评论 0 12
  • 课程介绍 先修课:概率统计,程序设计实习,集合论与图论 后续课:算法分析与设计,编译原理,操作系统,数据库概论,人...
    ShellyWhen阅读 2,377评论 0 3
  • 1.把二元查找树转变成排序的双向链表 题目: 输入一棵二元查找树,将该二元查找树转换成一个排序的双向链表。 要求不...
    曲终人散Li阅读 3,388评论 0 19
  • 明天就是一年一度的教师节了,相信许多在校学生已经在想明天要怎么给自己的恩师表白了,或者礼物都已准备好就等明天送出去...
    行走的小熊阅读 398评论 0 0
  • 一件从来没有给别人说过的事,现在差不多两周年。 她叫安聪睿,在我看来这个名字拗口的无法理喻。她说那你就叫我网名呗,...
    皮卡丘不会飞阅读 495评论 0 0