您的位置:首页 > 理论基础

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

2018-04-04 10:41 387 查看
总时间限制: 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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: