思路
蛮经典的一道题,然而一直没做过
总之就是用大根堆和小根堆来取代平衡树,由于询问的第
k
k
k 大是单调的,所以维护一个一直有
k
−
1
k-1
k−1 个元素的大根堆,对于每个新加的元素,若大根堆未满,则直接加入大根堆,否则判断它和大根堆堆顶大小,取二者中大的一个pop出并加入小根堆,这样保证大根堆元素数量为
k
−
1
k-1
k−1,并且大根堆堆顶大小不超过小根堆堆顶,于是小根堆堆顶即为第
k
k
k 大。
代码
这个输入好迷惑,一开始没看明白,直接存成数组了。
priority_queue<int> p1; //da
priority_queue<int, vector<int>, greater<int> > p2;//xiao
int n, m;
int a[N];
int u[N];
int main() {
cin >> n >> m;
for(int i = 1; i <= n; i++) {
cin >> a[i];
}
for(int i = 1; i <= m; i++)
cin >> u[i];
int l = 1;
for(int i = 1; i <= n; i++) {
if(l > m) break;
p1.push(a[i]);
if(p1.size() == l) {
p2.push(p1.top());
p1.pop();
}
while(u[l] == i && l <= m) {
cout << p2.top() << endl;
p1.push(p2.top());
p2.pop();
l++;
}
}
return 0;
}