LUOGUP1801 黑匣子_NOI導刊2010提高(06)
阿新 • • 發佈:2019-01-26
一道堆排。主要思路就是建兩個堆,一個大根堆,一個小根堆,大根堆存1-i的,小根堆存i+1-u[i]的。然後如果大根堆的堆頂小於了小根堆的堆頂說明大根堆不是1-i的了,就需要維護這兩個堆,交換他們的堆頂。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int m,n;
int t;
//int i,j;
int u[200005],a[200005],me[200005];
int hp[200006];
int cnt;
void pusx(int x) {
hp[++cnt]=x;
int now=cnt;
while(now>1) {
int tp=now/2;
if(hp[tp]<hp[now]) {
swap(hp[tp],hp[now]);
now=tp;
} else break;
}
}
void pus(int x) {
a[++t]=x;
int now=t;
while(now>1) {
int tp=now/2;
if(a[tp]>a[now]) {
swap(a[tp],a[now]);
now=tp;
} else break;
}
}
int del() {
int res=a[1];
a[1]=a[t];
int now=1;
t--;
while(2*now<=t) {
int tp=now*2;
if(tp<t&&a[tp]>a[tp+1])tp++;
if(a[tp]<a[now]) {
swap(a[tp],a[now]);
now=tp;
} else break;
}
return res;
}
int delx() {
int res=hp[1];
hp[1]=hp[cnt];
int now=1;
cnt--;
while(2*now<=cnt) {
int tp=now*2;
if(tp<cnt&&hp[tp]<hp[tp+1])tp++;
if(hp[tp]>hp[now]) {
swap(hp[tp],hp[now]);
now=tp;
} else break;
}
return res;
}
int main() {
memset(a,0x7f,sizeof a);
memset(hp,0x8f,sizeof hp);
scanf("%d%d",&m,&n);
for(register int i=1; i<=m; i++) {
scanf("%d",&me[i]);
}
for(register int i=1; i<=n; i++) {
scanf("%d",&u[i]);
}
for(register int i=1; i<=n; i++) {
for(register int j=u[i-1]+1; j<=u[i]; j++) {
pus(me[j]);//建小根堆
}
int k;
int r;
r=del();
pusx(r);//建大根堆
while(a[1] < hp[1] && t>0 && cnt>0) {//我認為的核心內容,維護這兩個堆
k=delx();
r=del();
pus(k);
pusx(r);
}
printf("%d\n",hp[1]);
}
}