洛谷 [P1801] 黑匣子
2017-10-28 08:18
197 查看
这道题是一道splay裸题,然而身为蒟蒻的我并不会,所以这道题我维护的是一个大根堆与一个小根堆结合起来的类似沙漏的结构。
本题难点在于询问的不是最大最小值,而是第K小值,所以我们想到了维护这样两个堆,上面是一个大小限定为K-1的大根堆,下面是一个小根堆,每次插入/查询操作时,保持前K-1大的始终在大根堆内。
插入/查询函数:
对整体的插入/查询:
附:跑得飞快的代码:
本题难点在于询问的不是最大最小值,而是第K小值,所以我们想到了维护这样两个堆,上面是一个大小限定为K-1的大根堆,下面是一个小根堆,每次插入/查询操作时,保持前K-1大的始终在大根堆内。
插入/查询函数:
int heap[200005][3],hsize[3]; int m,n,num[200005],temp; void put(int x,int i){ heap[++hsize[i]][i]=x; int son=hsize[i],pa=son>>1; while(pa>=1){ if(i==1){ if(heap[pa][i]>heap[son][i]){ int t=heap[pa][i];heap[pa][i]=heap[son][i];heap[son][i]=t; son=pa; pa>>=1; }else break; }else { if(heap[pa][i]<heap[son][i]){ int t=heap[pa][i];heap[pa][i]=heap[son][i];heap[son][i]=t; son=pa; pa>>=1; }else break; } } } int get(int i){ int rv=heap[1][i]; heap[1][i]=heap[hsize[i]--][i]; int pa=1,son; while(pa<=(hsize[i]>>1)){ if(i==1) son=heap[pa<<1][i]<heap[pa<<1|1][i]?pa<<1:pa<<1|1; else son=heap[pa<<1][i]>heap[pa<<1|1][i]?pa<<1:pa<<1|1; if(i==1){ if(heap[pa][i]>heap[son][i]){ int t=heap[pa][i];heap[pa][i]=heap[son][i];heap[son][i]=t; pa=son; }else break; }else { if(heap[pa][i]<heap[son][i]){ int t=heap[pa][i];heap[pa][i]=heap[son][i];heap[son][i]=t; pa=son; }else break; } } return rv; }
对整体的插入/查询:
void putt(int x){ put(x,1); if(x==heap[1][1]){ get(1); put(x,2); int t=get(2); put(t,1); } } int gett(){ temp++; int rv=get(1); put(rv,2); return rv; }
附:跑得飞快的代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cmath>
using namespace std;
int read(){
int rv=0,fh=1;
char c=getchar();
while(c<'0'||c>'9'){
if(c=='-') fh=-1;
c=getchar();
}
while(c>='0'&&c<='9'){
rv=(rv<<1)+(rv<<3)+c-'0';
c=getchar();
}
return rv*fh;
}
int heap[200005][3],hsize[3]; int m,n,num[200005],temp; void put(int x,int i){ heap[++hsize[i]][i]=x; int son=hsize[i],pa=son>>1; while(pa>=1){ if(i==1){ if(heap[pa][i]>heap[son][i]){ int t=heap[pa][i];heap[pa][i]=heap[son][i];heap[son][i]=t; son=pa; pa>>=1; }else break; }else { if(heap[pa][i]<heap[son][i]){ int t=heap[pa][i];heap[pa][i]=heap[son][i];heap[son][i]=t; son=pa; pa>>=1; }else break; } } } int get(int i){ int rv=heap[1][i]; heap[1][i]=heap[hsize[i]--][i]; int pa=1,son; while(pa<=(hsize[i]>>1)){ if(i==1) son=heap[pa<<1][i]<heap[pa<<1|1][i]?pa<<1:pa<<1|1; else son=heap[pa<<1][i]>heap[pa<<1|1][i]?pa<<1:pa<<1|1; if(i==1){ if(heap[pa][i]>heap[son][i]){ int t=heap[pa][i];heap[pa][i]=heap[son][i];heap[son][i]=t; pa=son; }else break; }else { if(heap[pa][i]<heap[son][i]){ int t=heap[pa][i];heap[pa][i]=heap[son][i];heap[son][i]=t; pa=son; }else break; } } return rv; }
void putt(int x){ put(x,1); if(x==heap[1][1]){ get(1); put(x,2); int t=get(2); put(t,1); } } int gett(){ temp++; int rv=get(1); put(rv,2); return rv; }
int f[200005];
int main(){
freopen("in.txt","r",stdin);
m=read();n=read();
for(int i=1;i<=m;i++){
num[i]=read();
}
for(int i=1;i<=n;i++){
f[read()]++;
}
for(int i=1;i<=m;i++){
putt(num[i]);
while(f[i]){
printf("%d\n",gett());
f[i]--;
}
}
/*for(int i=1;i<=10000;i++){
put(rand(),2);
}
for(int i=1;i<=10000;i++){
printf("%d ",get(2));
}*/
fclose(stdin);
return 0;
}
相关文章推荐
- 「洛谷 P1801」黑匣子
- 洛谷 P1801 黑匣子_NOI导刊2010提高(06)
- 洛谷P1801 黑匣子_NOI导刊2010提高(06)
- 洛谷P1801 黑匣子
- 洛谷 P1801 黑匣子_NOI导刊2010提高(06)
- 洛谷 P1801 黑匣子_NOI导刊2010提高(06)
- 洛谷 P1801 [NOI导刊2010提高(06)] 黑匣子
- 洛谷 P1801 黑匣子_NOI导刊2010提高(06)
- P1801 黑匣子_NOI导刊2010提高(06)
- 堆or优先队列(洛谷1801 黑匣子_NOI导刊2010提高(06))
- 洛谷Oj-P1801 黑匣子-STL
- P1801 黑匣子(对顶堆)
- [P1801]黑匣子
- P1801 黑匣子_NOI导刊2010提高(06)
- 洛谷1801 黑匣子_NOI导刊2010提高(06) 【好题】
- |洛谷|NOI导刊|堆|黑匣子_NOI导刊2010提高(06)
- 洛谷Oj-P1801 黑匣子-堆
- 【洛谷1801】黑匣子
- 洛谷P1631 序列合并
- 洛谷—— P1640 [SCOI2010]连续攻击游戏