堆的实现

手写堆与C++优先队列
int heap[maxn], sz = 0;
void push(int x)
{
    int i = sz++; //自己节点的编号
    while (i > 0){
        int p = (i - 1) / 2;  //父亲节点的编号
        if (heap[p] <= x)  //如果已经没有大小颠倒则退出
            break;
        heap[i] = heap[p];  //把父亲节点的数值放下来,而把自己提上去
        i = p;
    }
    heap[i] = x;
}


int pop()
{
    int ret = heap[0];   //最小值
    int x = heap[--sz];  //要提到根的数值
    //从下开始交换
    int i = 0;
    while (i * 2 + 1 < sz){   //比较儿子的值
        int a = i * 2 + 1, b = i * 2 + 2;
        if (b < sz && heap[a] < heap[b])
            a = b;
        if (heap[a] >= x)   //如果已经没有大小颠倒则退出
            break;
        heap[i] = heap[a];   //把儿子的数值提上来
        i = a;
    }
    heap[i] = x;
    return ret;
}




实际上,大部分情况下并不需要自己实现堆。例如在C++中,STL里的priority_queue就是其中之一,不过需要注意的是,
priority_queue取出数值时得到的是最大值。


#include <queue>
#include <cstdio>
using namespace std;


int main()
{
    priority_queue<int> pque;     //声明
    pque.push(3);   //插入元素
    pque.push(5);
    pque.push(1);
    while (!pque.empty()){    //不断循环直到空为止
        printf("%d\n", pque.top());  //获取并删除最大值
        pque.pop();
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值