洛谷 P3378 【模板】堆

最新推荐文章于 2024-07-13 13:12:09 发布

S atur

最新推荐文章于 2024-07-13 13:12:09 发布

阅读量225

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/Satur9/article/details/104116673

版权

数据结构专栏收录该内容

49 篇文章

订阅专栏

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

#include<iostream>
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;
}