优先队列练习 黑匣子

1233: [Heap]黑匣子

时间限制: 1 Sec  内存限制: 64 MB
提交: 680  解决: 194
[提交][状态][讨论版]

题目描述

我们使用黑匣子的一个简单模型。它能存放一个整数序列和一个特别的变量i。在初始时刻，黑匣子为空且i等于0。这个黑匣子执行一系列的命令。有两类命令： (1)ADD(x)：把元素x放入黑匣子； (2)GET：i增1的同时，输出黑匣子内所有整数中第i小的数。牢记第i小的数是当黑匣子中的元素以非降序排序后位于第i位的元素(输入数据保证每次GET命令中第i小的数必存在)

输入

输入的第1行只有一个整数n,表示有n条命令. 第2行至第n+1行,每行一条命令.

输出

输出中有若干行内容,每行一个整数,表示相应的get命令的输出.

样例输入

11
ADD(3)
GET
ADD(1)
GET
ADD(-4)
ADD(2)
ADD(8)
ADD(-1000)
GET
GET
ADD(2)

样例输出

3
3
1
2

提示

对30%的数据,1≤n≤50000; 对100%的数据,1≤n≤500000; 对所有的数据,ADD命令输入的数在-10000至10000之间;

动态第k小？平衡树乱搞么。。。

查询的第k小中的k是单调递增的，也就是说只要维护前k个数就行了。

怎么维护？沙漏堆呗。

开一个大根堆，一个小跟堆，假设堆是一个三角形的样子，堆顶是某个顶点（算了还是画下图吧）。

对于上面那个大根堆，只要保证它的大小一直是k就行，由于k是单调增的，所以不会涉及到一些神奇的问题。

至于怎么维护大根堆的大小为k，自己想应该能想出来。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
#include <cmath>
#include <vector>
#include <map>
#include <string>
 
using namespace std;
 
char s[1000];
 
priority_queue<int> q1;
priority_queue<int, vector<int>, greater<int> > q2;
 
int n, a, p, pp;
 
char get(){
    return s[p ++];
}
 
void rd(int &x){
    x = 0;
    char c;
    int f = 1;
     
    char ret = 0;
     
    while((c = get() - '0'), (c > 9 || c < 0 || c == '-' - '0')){
        if(c == '-' - '0') f = -1;
        if(!ret && c >= 'A' && c <= 'Z') ret = c;
    }
    while((c >= 0 && c <= 9)) (x *= 10) += c, c = get() - '0';
    x *= f;
}
 
int main(){
    scanf("%d", &n);
    while(n --){
        scanf("%s", s);
        p = 0;
        rd(a);
//      cout << "A = " << a << endl;
        if(s[0] == 'A'){
//          cout << "DOA" << endl;
            if(q1.empty() || a < q1.top()){
                q1.push(a);
            }else{
                q2.push(a);
            }
        }else{
            pp ++;
            while(q1.size() && q1.size() > pp){
                q2.push(q1.top());
                q1.pop();
            }
            while(q1.size() < pp){
                q1.push(q2.top());
                q2.pop();
            }
            printf("%d\n", q1.top());
        }
    }
}