【优先队列】2017计算机学科夏令营上机考试 G:实现堆结构

总时间限制: 3000ms内存限制: 65536kB    题目链接——>点击打开链接    提交链接——>点击打开链接

描述

定义一个数组，初始化为空。在数组上执行两种操作：
1、增添1个元素，把1个新的元素放入数组。
2、输出并删除数组中最小的数。

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

输入

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

输出

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

样例输入

2
5
1 1
1 2
1 3
2
2
4
1 5
1 1
1 7
2

样例输出

1
2
1

提示

每组测试数据的复杂度为O(nlgn)的算法才能通过本次，否则会返回TLE(超时)

需要使用最小堆结构来实现本题的算法

题型/思路

     优先队列的应用。注意优先队列是用堆实现的，所以对优先队里push()、pop()操作的时间复杂度都是O（nlongn）

需要温习的是：priority_queue<int, vector<int>, greater<int> > p;   

初始化优先队列需要三个参数，第一个参数是元素类型，第二个参数为容器类型，第三个参数是比较算子。

还有自定义struct中operator的写法，return a>b表示优先输出值小的。

AC代码

#include<iostream>  
#include<algorithm>  
#include<queue>  
using namespace std;  

struct cmp{
	bool operator()(int a,int b){
		return a>b;//优先输出值小的 
	}
};

int main()  
{   
    priority_queue<int,vector<int>,cmp>pq;
    int T;
    cin>>T;
    while(T--){
    	int n;
    	cin>>n;
    	int flag;
    	for(int i=0;i<n;i++){
    		cin>>flag;
    		int num;
    		if(flag==1){
                cin>>num;
    			pq.push(num);
			}
			else if(flag==2){
				int tp=pq.top();
				pq.pop();
				//cout<<"删除"<<tp<<endl;
				cout<<tp<<endl;
			}
		}
	}
    
   
    return 0;  
}