luogu3378 堆

luogu3378 堆

时空限制    1000ms/128MB

题目描述

如题，初始小根堆为空，我们需要支持以下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的）

样例说明：

故输出为2、5

 

 

 

代码

法一：STL优先队列

#include<iostream>
#include<queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> > pq;

int main(){
	ios::sync_with_stdio(false);
	int n;
	cin>>n;
	for (int i=1,x,y; i<=n; i++){
		cin>>x;
		if (x==1) { cin>>y; pq.push(y); }
		else if (x==2) cout<<pq.top()<<endl;
		else pq.pop();
	}
	return 0;
}

法二：手动堆

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1000005;
int n,a[N],hsize=0;

void adjust1(int m){	//调整为小顶堆
	for (int i=m,j=i/2; i>1; )	//向上调整
		if (a[i]<a[j]){	//a[i]孩子节点  a[j]父亲节点
			swap(a[i],a[j]);
			i = j;
			j = i/2;
		}
		else break;
}

void adjust2(int i,int m){	//调整为小顶堆
	for (int j=2*i; j<=m; ){	//向下调整
		if (j<m && a[j+1]<a[j]) j++;	//存在右孩子且值更小
		if (a[i]>a[j]){	//a[i]父亲节点  a[j]孩子节点
			swap(a[i],a[j]);
			i = j;
			j = 2*i;
		}
		else break;
	}
}

int main(){
	ios::sync_with_stdio(false);
	cin>>n;
	for (int i=1,x; i<=n; i++){
		cin>>x;
		if (x==1) { cin>>a[++hsize]; adjust1(hsize); }
		else if (x==2) cout<<a[1]<<endl;
		else { a[1]=a[hsize--]; adjust2(1,hsize); }
	}
	return 0;
}