hihocoder1105 堆

堆是一种完全二叉树，分为大根堆和小根堆，以大根堆为例：

大根堆满足性质：对于堆中的任意一个节点，其值在以该节点为根的子树中最大。

支持的操作：O(logn)的insert和O(logn)的pop.

（1）insert操作：将插入元素放到完全二叉树的最后一个节点之后的一个节点，然后从该节点开始向上不断维护堆的性质（若其父亲节点的值比其小，则交换，直到到达根节点或者父节点的值不比其小）；

（2）pop操作：先将返回值（根节点的值）暂存，将最后一个节点的值赋给根节点，然后从根节点开始，向下维护堆的性质（若根节点值比其值最大的儿子的值小，那么与其交换，直到到达叶子节点或者其值不比值最大的儿子的值小），最后返回。

我的代码：

#include <iostream>
#include <string>

using namespace std;

#define MAXN 100005

struct bigRootHeap
{
    int t[MAXN], sz;
    void clear(){sz = 0;}
    void push(int x)
    {
        t[++sz] = x;
        int i = sz;
        while(i!=1&&t[i/2]<t[i])
        {
            swap(i/2, i);
            i = i/2;
        }
    }
    int pop()
    {
        if(sz==0) return 0;
        int ans = t[1];
        t[1] = t[sz--];
        int i = 1;
        while(2*i<=sz&&t[i]<t[bigSon(i)])
        {
            int j = bigSon(i);
            swap(i, j);
            i = j;
        }
        return ans;
    }
private:
    void swap(int i, int j)
    {
        int tmp = t[i];
        t[i] = t[j];
        t[j] = tmp;
    }
    int bigSon(int i)
    {
        if(2*i+1>sz) return 2*i;
        return t[2*i]>t[2*i+1]?2*i:2*i+1;
    }
}heap;

int main()
{
    int n;
    while(cin>>n)
    {
        heap.clear();
        while(n--)
        {
            string op;
            cin>>op;
            if(op[0]=='A')
            {
                int x;
                cin>>x;
                heap.push(x);
            }
            else    cout<<heap.pop()<<endl;
        }
    }
    return 0;
}

题目链接：http://hihocoder.com/problemset/problem/1105