主席树
一发主席树解决。
插入操作十分显然。
撤销操作复制前面的版本就行。
询问操作十分显然。
#include<iostream> #include<cstdio> #include<cstring> using namespace std; #define N 1000005 int n,lc[N],rc[N],rt[N],len[N],tot,k; char a[N],p[3],z[3]; #define mid (l+r)/2 void update(int &o,int pre,int x,int l,int r){ o=++tot; if(l>=r){scanf("%s",z);a[o]=z[0];return ;} lc[o]=lc[pre]; rc[o]=rc[pre]; if(x<=mid) update(lc[o],lc[pre],x,l,mid); else update(rc[o],rc[pre],x,mid+1,r); } void ask(int o,int x,int l,int r){ if(l>=r) {printf("%c\n",a[o]);return ;} if(x<=mid) ask(lc[o],x,l,mid); else ask(rc[o],x,mid+1,r); } int main(){ scanf("%d",&n); for(int i=1,q;i<=n;++i){ scanf("%s",p); if(p[0]==‘T‘){ ++k;len[k]=len[k-1]+1; update(rt[k],rt[k-1],len[k],1,n); }else if(p[0]==‘Q‘) scanf("%d",&q),ask(rt[k],q,1,n); else if(p[0]==‘U‘){ scanf("%d",&q);++k; len[k]=len[k-q-1];rt[k]=rt[k-q-1];//复制版本 } }return 0; }
原文地址:https://www.cnblogs.com/kafuuchino/p/10339700.html
时间: 2024-09-29 17:35:50