题目链接:

https://www.luogu.org/problem/show?pid=1801

题解:

堆的巧妙应(ru)用(men)
由于所有询问第 <nobr> k </nobr>小值的k是不断递增的,所以考虑可以使用堆来维护
建立一个大根堆和一个小根堆,用大根堆来维护前k小的数,大根堆中数的个数等于就应该等于k,此时堆顶就应该是第k小的数
加一个数时,把该数加入大根堆,如果大根堆的大小大于k,不断弹出堆顶加进小根堆
每次查询时直接访问大根堆的堆顶
如果要查询的第k小值的k增加,则将小根堆堆顶不断弹出,加入到大根堆中即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define reg register int
using namespace std;
int n,m;

priority_queue<int> q1;
priority_queue<int,vector<int>,greater<int> > q2;
int a[200010],b[200010];


int main()
{
    scanf("%d%d",&n,&m);
    for(reg i=1;i<=n;i++) scanf("%d",&a[i]);
    for(reg i=1;i<=m;i++) scanf("%d",&b[i]);
    sort(b+1,b+m+1);
    int pos=1;
    for(int i=1;i<=n;i++)
    {
        q1.push(a[i]);
        if(q1.size()>pos)
        {
            q2.push(q1.top());
            q1.pop();
        }
        while(b[pos]==i)
        {
            pos++;
            printf("%d\n",q1.top());
            if(!q2.empty())
            {
                q1.push(q2.top());
                q2.pop();
            }
        }    
    }
    return 0;
}