优先级队列-堆

#include <cstdio>
#include <iostream>

using namespace std;

const int MAX_N = 1000;

// 用数组来实现二叉树
// 左儿子编号=自己*2 + 1
// 右儿子编号=自己*2 + 1
int heap[MAX_N],sz=0;

// 实现最小堆
void push(int x)
{
    // 取最后一个节点的编号，并递增。
    int i=sz++;
    // 循环更改所有颠倒序列
    while(i>0)
    {
        int p=(i-1)/2;
        // 无颠倒序列，退出循环
        if(heap[p]<=x)
        {
            break;
        }
        else
        {
            heap[i]=heap[p];
            i=p;
        }
    }
    // 填入新数
    heap[i]=x;
}

// 返回最小值（堆顶），然后在堆顶放置堆中最后一个元素
// 向下更新，直到有序
int pop()
{
    int ret=heap[0];
    // 记录要提到根节点的数
    int x=heap[--sz];
    int i=0;
    while(2*i+1 < sz)
    {
       int a=2*i+1,b=2*i+2;
       // 如果存在右儿子，且右儿子的值比做儿子小，则记左右儿子中较小者编号a为右儿子的编号
       if(b<sz && heap[a]>heap[b])
       {
           a=b;
       }
       // 无逆序，退出循环
       if(heap[a]>=x)
       {
           break;
       }
       // 否则将较小数前提,继续向下寻找
       heap[i]=heap[a];
       i=a;
    }
    // 最后将x写入适当的位置
    heap[i]=x;

    return ret;
}


int main()
{
    push(1);
    push(5);
    push(4);

    cout<<pop()<<endl;
    push(2);
    push(7);
    cout<<pop()<<endl;
    cout<<pop()<<endl;
    push(9);
    for(int i=0;i<3;++i)
    {
        cout<<pop()<<endl;
    }
    return 0;
}