P1801 黑匣子 对顶堆

link

思路

蛮经典的一道题，然而一直没做过
总之就是用大根堆和小根堆来取代平衡树，由于询问的第 $k$ 大是单调的，所以维护一个一直有 $k-1$ 个元素的大根堆，对于每个新加的元素，若大根堆未满，则直接加入大根堆，否则判断它和大根堆堆顶大小，取二者中大的一个pop出并加入小根堆，这样保证大根堆元素数量为$k-1$，并且大根堆堆顶大小不超过小根堆堆顶，于是小根堆堆顶即为第 $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;
}