洛谷 P3378 【模板】堆

题目链接:https://www.luogu.com.cn/problem/P3378
初学选手的堆类模板题,这里就用数组模拟堆操作。
代码实现:

#include
using namespace std;

const int N = 1e6 + 5;

int n, size;
int h[N];

void down(int u)
{
    int t = u;
    if(u * 2 <= size && h[u * 2] < h[t]) t = u * 2;
    if(u * 2 + 1 <= size && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
    if(t != u){
        swap(h[u], h[t]);
        down(t);
    }
}

void up(int u)
{
    int t = u;
    if(u / 2 && h[u / 2] > h[t]){
        t = u / 2;
        swap(h[u], h[t]);
        up(t);
    }
}

int main()
{
    cin >> n;
    while(n -- ){
        int k, x;
        cin >> k;
        if(k == 1){
            cin >> x;
            size ++ ;
            h[size] = x;
            up(size);
        }
        else if(k == 2)
            cout << h[1] << endl;
        else{
            h[1] = h[size];
            size -- ;
            down(1);
        }
    }
    return 0;
}

你可能感兴趣的:(数据结构)