正题
题目链接:
https://www.luogu.org/problemnew/show/P1801
大意
有两种操作,
1.将一个数放入黑匣子中
2.++i之后查询第i小的数(初始为i)
解题思路
维护两个堆,一个是比第i小的数大的数的最小堆,一个是比一个是比第i+1小的数小的数的最大堆,插入一个数时,若他比第二个堆的堆顶小就将第二个堆的堆顶放入第一个堆,然后将那个数放入第二个堆,不然就直接放入第一个堆,输出就直接输出第一个堆的堆顶。
代码
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,num,num2,a[200001],b[200001];
int u[200001],number[200001];
void up1(int x)
{
while (x>1 && a[x/2]>a[x])
{
swap(a[x/2],a[x]);
x/=2;
}
}//维护堆1
void down1(int x)
{
int y;
while (x*2<=num && a[x*2]<a[x] || x*2+1<=num && a[x*2+1]<a[x])
{
y=x*2;
if (y+1<=num && a[y]>a[y+1]) y++;
swap(a[y],a[x]);
x=y;
}
}//维护堆1
void up2(int x)
{
while (x>1 && b[x/2]<b[x])
{
swap(b[x/2],b[x]);
x/=2;
}
}//维护堆2
void down2(int x)
{
int y;
while (x*2<=num2 && b[x*2]>b[x] || x*2+1<=num2 && b[x*2+1]>b[x])
{
y=x*2;
if (y+1<=num2 && b[y]<b[y+1]) y++;
swap(b[y],b[x]);
x=y;
}
}//维护堆2
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
scanf("%d",&number[i]);
for (int i=1;i<=m;i++)
scanf("%d",&u[i]);
for (int i=1,j=1;i<=m;i++)
{
for (;j<=u[i];j++)//插入数
{
if (number[j]<b[1])
{
int z=b[1];
b[1]=number[j];
down2(1);
a[++num]=z;
up1(num);
}
else
{
a[++num]=number[j];
up1(num);
}
}
printf("%d\n",a[1]);//输出
b[++num2]=a[1];//去掉该数
up2(num2);
swap(a[1],a[num]);
num--;
down1(1);
}
}