优先级队列(二叉堆)

堆是一颗被完全填满的二叉树,底层的元素从左到右填入。
这里写图片描述
所以一个高度为h的堆有2^h到2^(h+1)-1个节点,这样的堆可以用一个数组来表示,堆顶为arr[0],父节点的下标为n,左儿子的下标就是2n+1,
堆有大堆(父节点的值大于儿子节点的值)和小堆(父节点的值小于儿子节点的值),大堆的根节点就是数组中最大的值,小堆的根节点就是数组中最小的值。堆只能访问根节点,能进行插入和删除操作。
这里写图片描述
一共10个元素从第(10-1)/2 = 4开始对每个元素排序一直到排到根节点(第n-1),每个节点的排序按照从上到下的方式如果父节点大于儿子节点中较小的,则交换,否则对下一个元素排序(小堆)

#include <iostream>
using namespace std;
#include <vector>

template <class T>
struct Less {
    bool operator()( const T& Left, const T& Right) //仿函数
    {
        return Left < Right;
    }
};

template <class T>
struct Greater {
    bool operator()(const T& Left, const T& Right)
    {
        return Left > Right;
    }
};

template <class T, class Compare = Less<T> > //默认为小堆
class Heap {
public:
    Heap() {};
    Heap(const T arr[], size_t sz)
    {
        assert(arr);
        for (int i = 0; i < sz; ++i)
            _heap.push_back(arr[i]);    //用容器存放元素
        if (sz > 1)
            for (int j = (sz - 1) / 2; j >= 0; --j)  
            {
                _AdjustDown(j);
            }

    }

    size_t Size()const
    {
        return _heap.size();
    }
    bool Empty()const
    {
        return _heap.empty();
    }

    void Insert(const T& data) //插入
    {
        _heap.push_back(data);
        if (Size() > 1)
            _AdjustUp(Size() - 1);
    }

    void Remove()   //删除堆顶 (交换第一个和最后一个,将最后一个节点删除,对堆顶向下排序)
    {
        if (Empty())
            return;
        if (Size() == 1)
            _heap.pop_back();
        else
        {
            std::swap(_heap[0], _heap[Size() - 1]);
            _heap.pop_back();
            _AdjustDown(0);
        }
    }
protected:
    void _AdjustDown(size_t parent)
    {
        size_t chlid = 2 * parent + 1;
        while (chlid < _heap.size())
        {
            if (chlid + 1 < _heap.size())
                chlid = ( Compare()(_heap[chlid], _heap[chlid+1])) ? chlid : (chlid + 1);
            if (Compare()(_heap[parent], _heap[chlid]) )
                std::swap(_heap[parent], _heap[chlid]);
            parent = chlid;
            chlid = 2 * parent + 1;
        }
    }

    void _AdjustUp(size_t chlid)
    {
        size_t parent = (chlid - 1) / 2;
        while (chlid != 0)
        {
            if ( Compare()(_heap[chlid], _heap[parent]))
                std::swap(_heap[chlid], _heap[parent]);
            else
                break;

            chlid = parent;
            parent = (chlid - 1) / 2;
        }
    }


protected:
    std::vector<T> _heap;
};


template <class T, class Compare = Less<T>>  //用小堆构建优先级队列
class PriorityQueue {
public:
    PriorityQueue()
        :_hp()
    {}
    void Push(const T& data)
    {
        _hp.Insert(data);
    }

    void Pop()
    {
        _hp.Remove();
    }

    const T& Top()const
    {
        return _hp._heap[0];
    }

    size_t Size()const
    {
        return _hp.Size();
    }

    bool Empty()const
    {
        return _hp.Empty();
    }

protected:
    Heap<T, Compare > _hp;
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

魏尔肖

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值