最小堆的C++实现

写在前面

闲来无事写了个C++版本的最小堆,堆的底层采用数组,基本的上浮和下沉过程请参考算法导论,非常简单。这里不再画图解释,具体的实现代码如下。

代码实现

template<typename T>
class PriorityQueue {

private:
    vector<T> heap;
public:
    PriorityQueue() {
    }
    void push(T value) {
        heap.push_back(value);
        if(heap.size() == 1) return;
        int size = heap.size();
        auto pos = size-1;
        using std::swap;
        while(pos>0&&heap[(pos-1)/2]>heap[pos]) {
            swap(heap[(pos-1)/2],heap[pos]);
            pos = (pos-1)/2;
        }
    }
    T pop() {
        if(heap.size() == 0) return -1;
        auto top = heap[0];
        using std::swap;
        swap(heap[0],heap.back());
        heap.pop_back();
        int size = heap.size();
        int pos = 0;
        while(pos<size) {
            int left = pos*2+1;
            if(left>=size) break;
            if(left+1<size&&heap[left]>heap[left+1]) left++;
            if(heap[left]>heap[pos]) break;
            swap(heap[left],heap[pos]);
            pos = left;
        }
        return top;
    }
    vector<T>& getHeapVector(){return heap;};
    T top() {
        return heap[0];
    }
    int size() {return heap.size();};
};

该版本是非常简化的版本,只做参考。

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值