黑匣子
题目描述
Black Box 是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量 i i i。最开始的时候 Black Box 是空的.而 i = 0 i=0 i=0。这个 Black Box 要处理一串命令。
命令只有两种:
-
ADD(x)
:把 x x x 元素放进 Black Box; -
GET
: i i i 加 1 1 1,然后输出 Black Box 中第 i i i 小的数。
记住:第 i i i 小的数,就是 Black Box 里的数的按从小到大的顺序排序后的第 i i i 个元素。
我们来演示一下一个有11个命令的命令串。(如下表所示)
序号 | 操作 | i i i | 数据库 | 输出 |
---|---|---|---|---|
1 | ADD(3) | 0 0 0 | 3 3 3 | / |
2 | GET | 1 1 1 | 3 3 3 | 3 3 3 |
3 | ADD(1) | 1 1 1 | 1 , 3 1,3 1,3 | / |
4 | GET | 2 2 2 | 1 , 3 1,3 1,3 | 3 3 3 |
5 | ADD(-4) | 2 2 2 | − 4 , 1 , 3 -4,1,3 −4,1,3 | / |
6 | ADD(2) | 2 2 2 | − 4 , 1 , 2 , 3 -4,1,2,3 −4,1,2,3 | / |
7 | ADD(8) | 2 2 2 | − 4 , 1 , 2 , 3 , 8 -4,1,2,3,8 −4,1,2,3,8 | / |
8 | ADD(-1000) | 2 2 2 | − 1000 , − 4 , 1 , 2 , 3 , 8 -1000,-4,1,2,3,8 −1000,−4,1,2,3,8 | / |
9 | GET | 3 3 3 | − 1000 , − 4 , 1 , 2 , 3 , 8 -1000,-4,1,2,3,8 −1000,−4,1,2,3,8 | 1 1 1 |
10 | GET | 4 4 4 | − 1000 , − 4 , 1 , 2 , 3 , 8 -1000,-4,1,2,3,8 −1000,−4,1,2,3,8 | 2 2 2 |
11 | ADD(2) | 4 4 4 | − 1000 , − 4 , 1 , 2 , 2 , 3 , 8 -1000,-4,1,2,2,3,8 −1000,−4,1,2,2,3,8 | / |
现在要求找出对于给定的命令串的最好的处理方法。ADD
命令共有
m
m
m 个,GET
命令共有
n
n
n 个。现在用两个整数数组来表示命令串:
-
a 1 , a 2 , ⋯ , a m a_1,a_2,\cdots,a_m a1,a2,⋯,am:一串将要被放进 Black Box 的元素。例如上面的例子中 a = [ 3 , 1 , − 4 , 2 , 8 , − 1000 , 2 ] a=[3,1,-4,2,8,-1000,2] a=[3,1,−4,2,8,−1000,2]。
-
u 1 , u 2 , ⋯ , u n u_1,u_2,\cdots,u_n u1,u2,⋯,un:表示第 u i u_i ui 个元素被放进了 Black Box 里后就出现一个
GET
命令。例如上面的例子中 u = [ 1 , 2 , 6 , 6 ] u=[1,2,6,6] u=[1,2,6,6] 。输入数据不用判错。
输入格式
第一行两个整数
m
m
m 和
n
n
n,表示元素的个数和 GET
命令的个数。
第二行共 m m m 个整数,从左至右第 i i i 个整数为 a i a_i ai,用空格隔开。
第三行共 n n n 个整数,从左至右第 i i i 个整数为 u i u_i ui,用空格隔开。
输出格式
输出 Black Box 根据命令串所得出的输出串,一个数字一行。
样例 #1
样例输入 #1
7 4
3 1 -4 2 8 -1000 2
1 2 6 6
样例输出 #1
3
3
1
2
提示
数据规模与约定
- 对于 30 % 30\% 30% 的数据, 1 ≤ n , m ≤ 1 0 4 1 \leq n,m \leq 10^{4} 1≤n,m≤104。
- 对于 50 % 50\% 50% 的数据, 1 ≤ n , m ≤ 1 0 5 1 \leq n,m \leq 10^{5} 1≤n,m≤105。
- 对于 100 % 100\% 100% 的数据, 1 ≤ n , m ≤ 2 × 1 0 5 , ∣ a i ∣ ≤ 2 × 1 0 9 1 \leq n,m \leq 2 \times 10^{5},|a_i| \leq 2 \times 10^{9} 1≤n,m≤2×105,∣ai∣≤2×109,保证 u u u 序列单调不降。
这道题大概意思就是,给你一段前缀区间,让你求出第k大的数是哪个
并且k单调递增。
刚开始没有想到思路,
后面看题解然后就发现有这种神奇的做法。
我们知道堆是一种插入删除都是logn操作的数据结构。
它通常只能访问最大或者最小值。
有人想 :这不是大材小用吗?
直接用普通变量维护不就行了吗。
但是对于第k大问题无能为力,
首先需要三个堆,一个大根堆,一个小根堆,一个只有一个元素的当前堆
首先用一个最大根保存所以比当前元素小的数。
第k大,那么最大根数量为k-1.
除此之外,我们还需要一个小根堆,小根堆的数不小于当前数,小根堆的数可以看成备用数,动态调整当前堆以及小根堆。
接下来就是更新操作
对于新的一个数,我们对它与当前堆进行判断
如果大于等于当前堆,放入小根堆中。
否则与大根堆比对,总之要满足大根堆严格小于当前堆小于小根堆
代码如下
#include<bits/stdc++.h>
#include<algorithm>
using namespace std;
#define ll long long
const ll inf=5e9;
int a[200010],b[200010];
int main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<=m;i++)cin>>b[i];
priority_queue<int>h1;
priority_queue<int,vector<int>,greater<int> >h2;
int inx=1;
ll now;
for(int i=1;i<=n;i++){
if(i==1)now=a[i];
else {
if(now!=inf&&a[i]>=now){
h2.push(a[i]);
}
else {
if(h1.empty()==true||(a[i]>=h1.top()&&h1.empty()==false)){
h2.push(a[i]);
if(now!=inf) h2.push(now);
now=h2.top();
h2.pop();
}
else {
h1.push(a[i]);
h2.push(h1.top());
h1.pop();
if(now!=inf) h2.push(now);
now=h2.top();
h2.pop();
}
}
}
while(inx<=m&&b[inx]==i){
cout<<now<<endl;
h1.push(now);
if(h2.empty()==true) now=inf;
else {
now=h2.top();
h2.pop();
}
inx++;
}
}
return 0;
}
记得点个赞哦!!,码字不易