主席樹,查找第k位的數
op== 1,上一個狀態在末尾加一個點
op== 2,退回到前p個狀態,如果現在爲now,那麼退回now-p-1
op== 3,查詢列表中排k位的
注意主席樹範圍要開18倍或20倍
ACcode
#include<bits/stdc++.h>
#define re register
#define in Read()
using namespace std;
inline int in{
int i=0,f=1;char ch;
while(!isdigit(ch)&&ch!='-')ch=getchar();
if(ch=='-')ch=getchar(),f=-1;
while(isdigit(ch))i=(i<<1)+(i<<3)+ch-48,ch=getchar();
return i*f;
}
const int NNN=2e5+10;
int T,n,sz,now;
int rt[NNN];
struct Tree{
int sz;
int lson,rson;
int wei;
}tr[NNN*20];
inline void update(int pre,int &rt,int l,int r,int k,int w){
rt=++sz;
tr[rt]=tr[pre];
++tr[rt].sz;
if(l==r){
tr[rt].wei=w;
return;
}
int mid=(l+r)>>1;
if(k<=mid)update(tr[pre].lson,tr[rt].lson,l,mid,k,w);
else update(tr[pre].rson,tr[rt].rson,mid+1,r,k,w);
return;
}
inline int query(int p,int l,int r,int k){
if(l==r)return tr[p].wei;
int mid=(l+r)>>1;
if(k<=mid)query(tr[p].lson,l,mid,k);
else query(tr[p].rson,mid+1,r,k);
}
int main(){
T=in;
while(T--){
n=in,sz=0,now=0;
for(re int i=1;i<=n;++i){
int op=in,x=in;
if(op==1)
++now,
update(rt[now-1],rt[now],1,n,tr[rt[now-1]].sz+1,x);
if(op==2)
++now,
rt[now]=rt[now-x-1];
if(op==3)
printf("%d\n",query(rt[now],1,n,x));
}
}
return 0;
}