[主席树] 200606T2 Seq

传送门


主席树，查找第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;
}