JZOJ 5933. 【NOIP2018模擬10.27】百鴿籠
阿新 • • 發佈:2018-11-01
題目
鴿鴿鴿
有一個序列,有3種操作。
①刪除最左邊的元素
②在序列的最左邊加入一個數。
③查詢序列中第l個數到第r個數中第k小的數。
題解
想到之前ifk通過動態樹最終狀態的dfs序將問題轉化成靜態樹問題,我打算也用類似的方法。
其實不需要。
每次刪除和插入的都是最左邊的。難道不能將序列顛倒一下嗎?
模板題。
心得
主席樹不需要刪除元素,只需要選擇正確的rt[i],在rt[i]的基礎上加元素即可。
PS:將相同大小的陣列放在程式碼的同一行!!!不然很容易開小陣列
每次開陣列的時候留個心眼
程式碼
#include<iostream>
#include <cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 200010
#define M 400010
#define P(a) putchar(a)
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
struct note{
int ls,rs,sum;
}tr[M*20];
struct noq{
int l,r,k,op;
}qu[N];
int gs,rt[M];
int i,j,k,l,n,m,opx,d2,top,ans;
int nl,nr;
int o[M],a[M],t2,cx;
int c[N];
int read(){
int fh=0,rs=0;char ch=0;
while((ch<'0')||(ch>'9')&&(ch^'-'))ch=getchar();
if(ch=='-')fh=1,ch=getchar();
while(ch>='0'&&ch<='9')rs=(rs<<3)+(rs<<1)+(ch^'0' ),ch=getchar();
return fh?-rs:rs;
}
void write(int x){
if(x>9)write(x/10);
P(x%10+'0');
}
void change(int &px,int py,int l,int r){
px=++gs;
tr[px]=tr[py];
tr[px].sum++;
if(l==r)return;
int wz=(l+r)>>1;
if(opx<=wz)change(tr[px].ls,tr[py].ls,l,wz);
else change(tr[px].rs,tr[py].rs,wz+1,r);
}
int query(int px,int py,int l,int r,int k){
if(l==r)return a[l];
int wz=(l+r)>>1,x=tr[tr[px].ls].sum-tr[tr[py].ls].sum;
if(x<k)return query(tr[px].rs,tr[py].rs,wz+1,r,k-x);
else return query(tr[px].ls,tr[py].ls,l,wz,k);
}
int main(){
n=read();m=read();
fo(i,1,n){
c[i]=read();
a[++t2]=c[i];
}
fo(i,1,m){
qu[i].op=read();
if(qu[i].op==2){
qu[i].k=read();
a[++t2]=qu[i].k;
}
if(qu[i].op==3){
qu[i].l=read(),qu[i].r=read(),qu[i].k=read();
}
}
sort(a+1,a+t2+1);
cx=unique(a+1,a+t2+1)-a-1;
fo(i,1,n)c[i]=lower_bound(a+1,a+cx+1,c[i])-a;
fo(i,1,m)if(qu[i].op==2)qu[i].k=lower_bound(a+1,a+cx+1,qu[i].k)-a;
fd(i,n,1){
opx=c[i];
change(rt[i],rt[i+1],1,cx);
o[++top]=i;
}
d2=n;
fo(i,1,m){
if(qu[i].op==1){
top--;
}else
if(qu[i].op==2){
o[++top]=++d2;
opx=qu[i].k;
change(rt[d2],rt[o[top-1]],1,cx);
}else{
nl=o[top-qu[i].r];
nr=o[top-qu[i].l+1];
ans=query(rt[nr],rt[nl],1,cx,qu[i].k);
write(ans);
P('\n');
}
}
return 0;
}