题目大意 :
题目链接:https://www.luogu.org/problemnew/show/P1801
两种操作:
- ADD(x)ADD(x)ADD(x):加入xxx元素
- GETGETGET:第iii次GETGETGET操作输出第iii小的元素
思路:
有点像对顶堆啊。
每次维护一个大根堆,保证只保存最小的sss个数字。其中sss表示现在进行了sss次GETGETGET操作。
然后维护一个小根堆,把其他所有数字储存进去。这样每次询问完就可以把不在大根堆里的最小的元素扔进大根堆里。方便下一次询问。
时间复杂度O(mlogn)O(m\log n)O(mlogn)
代码:
#include <queue>
#include <cstdio>
using namespace std;
const int N=200010;
int n,m,s,add[N],cnt[N];
priority_queue<int> q1,q2;
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
scanf("%d",&add[i]);
for (int i=1;i<=m;i++)
{
int x;
scanf("%d",&x);
cnt[x]++;
}
for (int i=1;i<=n;i++)
{
q1.push(add[i]);
while (q1.size()>s)
{
q2.push(-q1.top());
q1.pop();
}
for (;cnt[i];cnt[i]--)
{
s++;
q1.push(-q2.top());
q2.pop();
printf("%d\n",q1.top());
}
}
return 0;
}