用最大堆实现优先队列(c++)

关于最大堆,最小堆的概念这里不再介绍。

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

template<typename T>
class PriorityQueue {
    private:
        vector<T> v;
        int size;
        const static int initCap = 100;
    private:
        int parent(int pos) {
            return pos/2;
        }

        int left(int pos) {
            return pos*2;
        }

        int right(int pos) {
            return pos*2 + 1;
        }
    public:
        PriorityQueue():
            size(0) {
                v = vector<T>(initCap);
            }

        bool enQueue(const T& t) {
            size++;
            v[size] = t;
            int i = size;
            //插入到尾部,一直上提到使当前堆符合最大堆性质为止
            while(i > 1 && v[i] > v[parent(i)]) {
                T tmp = v[i];
                v[i] = v[parent(i)];
                v[parent(i)] = tmp;
                i = parent(i);
            }
            return true;
        }

        bool deQueue(T& max) {
            //取出最大值,并把尾部元素和最大值交换,然后维护当前堆到符合堆性质
            if(size < 1) {
                return false;
            }
            max = v[1];
            v[1] = v[size];
            v[size] = max;
            size--;
            maxHeapIfy(1);
            return true;
        }

        void maxHeapIfy(int pos) {
            int largestPos = pos;
            if(left(pos) <= size && v[pos] < v[left(pos)]) {
                largestPos = left(pos);
            }
            if(right(pos) <= size && v[largestPos] < v[right(pos)]) {
                largestPos = right(pos);
            }
            if(largestPos != pos) {
                T tmp = v[pos];
                v[pos] = v[largestPos];
                v[largestPos] = tmp;
                maxHeapIfy(largestPos);
            }
        }

        int getSize() {
            return size;
        }

        bool top(T& t) {
            if(size < 1) {
                return false;
            }
            t = v[1];
            return true;
        }
};

int main() {
    PriorityQueue<int> pQ;
    pQ.enQueue(8);
    pQ.enQueue(1);
    pQ.enQueue(3);
    pQ.enQueue(6);
    pQ.enQueue(5);
    pQ.enQueue(3);
    pQ.enQueue(10);
    pQ.enQueue(9);
    pQ.enQueue(7);
    int a = -1;
    while(pQ.deQueue(a)) {
        cout<<a<<" ";
    }
    cout<<endl;
    return 0;
}
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值