lg1383

链接:lg1383

题解

  • 主席树水题
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define rep(i,x,y) for(register int i=x;i<=y;++i)
#define repd(i,x,y) for(register int i=x;i>=y;--i)
#define ll long long
using namespace std;
const int N=1e5+7;
const int M=1e5;
char rt[N<<5];
int lson[N<<5],rson[N<<5],root[N],cnt,now,len[N],n;
inline void update(int pre,int &o,int l,int r,int k,char x){
o=++cnt;
lson[o]=lson[pre];rson[o]=rson[pre];
if(l==r){rt[o]=x; return ;}
int mid=l+r>>1;
if(k<=mid)update(lson[pre],lson[o],l,mid,k,x);
else update(rson[pre],rson[o],mid+1,r,k,x);
}
inline void query(int o,int l,int r,int k){
if(l==r){cout<<rt[o]<<endl;return ;}
int mid=l+r>>1;
if(k<=mid)return query(lson[o],l,mid,k);
return query(rson[o],mid+1,r,k);
}
int main(){
ios::sync_with_stdio(false);
cin>>n;
while(n--){
char op,c;int x;
cin>>op;
if(op=='T'){cin>>c;
root[++now]=root[now-1];
len[now]=len[now-1]+1;
update(root[now-1],root[now],1,M,len[now],c);
}
else if(op=='U'){cin>>x;root[++now]=root[now-x-1];len[now]=len[now-x-1];}
else {cin>>x;query(root[now],1,M,x);}
}
return 0;
}