C++之路进阶——codevs3333（高级打字机）

3333 高级打字机

 时间限制: 1 s

 空间限制: 256000 KB

 题目等级 : 大师 Master

 

 

题目描述 Description

早苗入手了最新的高级打字机。最新款自然有着与以往不同的功能，那就是它具备撤销功能，厉害吧。

请为这种高级打字机设计一个程序，支持如下3种操作：

1.T x：在文章末尾打下一个小写字母x。(type操作)

2.U x：撤销最后的x次修改操作。（Undo操作）

（注意Query操作并不算修改操作）

3.Q x：询问当前文章中第x个字母并输出。（Query操作）

文章一开始可以视为空串。

输入描述 Input Description

第1行：一个整数n，表示操作数量。

以下n行，每行一个命令。保证输入的命令合法。

输出描述 Output Description

每行输出一个字母，表示Query操作的答案。

样例输入 Sample Input

7

T a

T b

T c

Q 2

U 2

T c

Q 2

样例输出 Sample Output

b

c

数据范围及提示 Data Size & Hint

对于40%的数据 n<=200;

对于50%的数据 n<=100000；保证Undo操作不会撤销Undo操作。

<高级挑战>

对于100%的数据 n<=100000；Undo操作可以撤销Undo操作。

题解：

可持久化线段树维护新加入的字符。遇到撤销操作直接转到root[cnt-x-1]的线段树上。

 #include<cstdio>
 #include<iostream>
 #include<algorithm>
 #define maxn 2000000

 using namespace std;

 int root[maxn],ls[maxn],rs[maxn],len[maxn],n,x,cnt,tot;

 char S[maxn<<],s[],ch[];

 void update(int &y,int x,int l,int r,int pos,char c)
     {
       y=++tot;
       if (l==r) {S[y]=c;return;}
       ls[y]=ls[x],rs[y]=rs[x];
       int mid=(l+r)>>;
       if (pos<=mid) update(ls[y],ls[x],l,mid,pos,c);
           else update(rs[y],rs[x],mid+,r,pos,c);
     }

 char query(int rt,int l,int r,int pos)
     {
          if (l==r) {return S[rt];}
          int mid=(l+r)>>;
          if (pos<=mid) return query(ls[rt],l,mid,pos);
          if (pos>mid) return query(rs[rt],mid+,r,pos);
     }

 int main()
    {
         scanf("%d",&n);
         for (int i=;i<=n;i++)
            {
               scanf("%s",s);
            if (s[]=='T')
              {
                  cnt++;
                  len[cnt]=len[cnt-]+;
                  scanf("%s",ch);
                  update(root[cnt],root[cnt-],,,len[cnt],ch[]);
              }
            if (s[]=='U')
              {
                  scanf("%d",&x);
                  cnt++;
                  root[cnt]=root[cnt-x-];
                  len[cnt]=len[cnt-x-];
              }
            if (s[]=='Q')
              {
                  scanf("%d",&x);
                  cout<<query(root[cnt],,,x)<<endl;
              }
            }
        return ;
    }