正題

題目鏈接：
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);} }

總結

以上是生活随笔為你收集整理的P1801-黑匣子_NOI导刊2010提高【堆】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。