洛谷-3378 【模板】堆

最新推荐文章于 2022-10-08 00:45:46 发布

原创最新推荐文章于 2022-10-08 00:45:46 发布 · 192 阅读

0 ·

CC 4.0 BY-SA版权

算法题同时被 2 个专栏收录

311 篇文章

订阅专栏

模板

87 篇文章

订阅专栏

本文详细解析了小根堆的基本操作，包括插入、查找最小元素和删除最小元素。通过具体的代码实现，展示了如何处理多种堆操作，适用于初学者理解和掌握小根堆数据结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述
如题，初始小根堆为空，我们需要支持以下3种操作：
操作1： 1 x 表示将x插入到堆中
操作2： 2 输出该小根堆内的最小数
操作3： 3 删除该小根堆内的最小数
输入输出格式
输入格式：
第一行包含一个整数N，表示操作的个数
接下来N行，每行包含1个或2个正整数，表示三种操作，格式如下：
操作1： 1 x
操作2： 2
操作3： 3
输出格式：
包含若干行正整数，每行依次对应一个操作2的结果。

输入输出样例
输入样例#1：
5
1 2
1 5
2
3
2

输出样例#1：
2
5

说明
时空限制：1000ms,128M
数据规模：
对于30%的数据：N<=15
对于70%的数据：N<=10000
对于100%的数据：N<=1000000（注意是6个0。。。不过不要害怕，经过编者实测，堆是可以AC的）

#include<cstdio>
#include<iostream>
#define N 30001
using namespace std;
int heap_size=0,n=0;
int heap[N]={0};
void put(int d){
    int now=0,next=0;
    heap[++heap_size]=d;
    now=heap_size;
    while(now>1){
        next=now>>1;
        if(heap[now]>=heap[next]) return;
        swap(heap[now],heap[next]);
        now=next;
    }
}
int del(){
    int now=0,next=0,res=0;
    res=heap[1];
    heap[1]=heap[heap_size--];
    now=1;
    while(now*2<=heap_size){
        next=now*2;
        if(next<heap_size&&heap[next+1]<heap[next]) next++;
        if(heap[now]<=heap[next]) return res;
        swap(heap[now],heap[next]);
        now=next;
    }
    return res;
}
int get(){
    return heap[1];
}
void work(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        if(x==1){scanf("%d",&x);put(x);}
        else if(x==2) {printf("%d\n",get());}
        else del();
    }
}
int main(){
    work();
}

#include<cstdio>
#include<iostream>
#include<algorithm>
#define N 1000005
using namespace std;
int heap_size=0;
int heap[N]={0};
void put(int x){
    heap[++heap_size]=x;
    push_heap(heap+1,heap+1+heap_size,greater<int>());  
}
int get(){    
    return heap[1];   
}
void del(){ 
    pop_heap(heap+1,heap+1+heap_size,greater<int>());  
    heap_size--;    
}
int main(){
    int n;
    scanf("%d",&n);
    int x;
    for(int i=1;i<=n;i++){
        scanf("%d",&x);
        if(x==1) {scanf("%d",&x);put(x);}
        else if(x==2) {printf("%d\n",get());}
        else if(x==3) del();      
    }
}