洛谷P1801 黑匣子 双堆套路的使用

题意

题目链接

题解

这道题本可以用Treap暴力求解出来，但是不够优雅，因为没有充分利用到题目中给的条件，那就是要求的$ith$小的值的$i$是单调递增的。

我们用两个堆来维护，大顶堆和小顶堆。
大顶堆中的元素是排好序的前$i-1$个，小顶堆里的元素个数是排好序的后$n-i+1$个。
这样每次询问的时候，我们只需要输出小顶堆中的最小的元素就ok了。
下面看怎么维护这两个堆。
一开始所有的元素都往小顶堆里面插入。
当i由1变成2的时候，我们把小顶堆里的最小值扔到大顶堆里面去，始终保持大顶堆里的元素个数是$i-1$。
如果插入一个元素比大顶堆里最大的元素还要大（或者相等），就把这个元素扔到小顶堆里面。否则扔到大顶堆里面，并且把大顶堆里面最大的元素扔到小顶堆里面去。
这就是维护过程。

代码

#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
const int maxn = 1e6;
int a[maxn],mark[maxn];
int n,m;
int main(){
    scanf("%d%d",&m,&n);
    for(int i = 1;i <= m;++i)
        scanf("%d",&a[i]);
    for(int i = 0;i < n;++i){
        int tmp;scanf("%d",&tmp);
        mark[tmp]++;
    }
    priority_queue<int> Q1;
    priority_queue<int,vector<int>,greater<int> >Q2;
    int cut = 0;
    for(int i = 1;i <= m;++i){
        if(Q1.empty() || a[i] >= Q1.top())
            Q2.push(a[i]);
        else{
            Q1.push(a[i]);
            Q2.push(Q1.top());
            Q1.pop();
        }
        for(int j = 0;j < mark[i];++j){
            printf("%d\n",Q2.top());
            Q1.push(Q2.top());
            Q2.pop();
        }
    }
    return 0;
}