题意
题目链接
题解
这道题本可以用Treap暴力求解出来,但是不够优雅,因为没有充分利用到题目中给的条件,那就是要求的ithith小的值的ii是单调递增的。
我们用两个堆来维护,大顶堆和小顶堆。
大顶堆中的元素是排好序的前个,小顶堆里的元素个数是排好序的后n−i+1n−i+1个。
这样每次询问的时候,我们只需要输出小顶堆中的最小的元素就ok了。
下面看怎么维护这两个堆。
一开始所有的元素都往小顶堆里面插入。
当i由1变成2的时候,我们把小顶堆里的最小值扔到大顶堆里面去,始终保持大顶堆里的元素个数是i−1i−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;
}