实现堆结构

总时间限制: 

3000ms

 

内存限制: 

65535kB

描述

定义一个数组，初始化为空。在数组上执行两种操作：

1、增添1个元素，把1个新的元素放入数组。

2、输出并删除数组中最小的数。

使用堆结构实现上述功能的高效算法。

输入

第一行输入一个整数t，代表测试数据的组数。
对于每组测试数据，第一行输入一个整数n，代表操作的次数。
每次操作首先输入一个整数type。
当type=1，增添操作，接着输入一个整数u，代表要插入的元素。
当type=2，输出删除操作，输出并删除数组中最小的元素。
1<=n<=100000。

输出

每次删除操作输出被删除的数字。

#include <iostream>
#include <cstring>
using namespace std;
#define maxn 100005

long sz, heap[maxn];


void Insert(int x)
{
    int i = sz++;
    while (i > 0) {
        int p = (i - 1) >> 1;
        if (heap[p] > x) {
            heap[i] = heap[p];
            i = p;
        }
        else break;
    }
    heap[i] = x;
}

void GetTop()
{
    long ans = heap[0];
    printf("%ld\n", ans);
    long x = heap[--sz];
    long i = 0;
    while ((i << 1) + 1 < sz) {
        long left = (i << 1) + 1;
        long right = (i << 1) + 2;
        if (right < sz && heap[right] < heap[left])
            left = right;
        if (heap[left] >= x)break;
        heap[i] = heap[left];
        i = left;
    }
    heap[i] = x;
}

int main()
{
    int t, n;
    cin >> t;
    while (t--) {
        sz = 0;
        memset(heap, 0, sizeof(heap));
        cin >> n;
        while (n--) {
            int type;
            long u;
            cin >> type;
            if (type == 1) {
                cin >> u;
                Insert(u);
            }
            else if (type == 2) 
                GetTop();
        }
    }
    //system("pause");
    return 0;
}