可持久化数组(结构体实现)

如题，你需要维护这样的一个长度为 N 的数组，支持如下几种操作
在某个历史版本上修改某一个位置上的值
访问某个历史版本上的某一位置的值
此外，每进行一次操作（对于操作2，即为生成一个完全一样的版本，不作任何改动），就会生成一个新的版本。版本编号即为当前操作的编号（从1开始编号，版本0表示初始状态数组）。

输入
输入的第一行包含两个正整数 N M， 分别表示数组的长度和操作的个数。
第二行包含 N 个整数，依次为初始状态下数组各位的值（依次为 ai ，1 ≤ i ≤ N）。
接下来 M行每行包含 3 或 4 个整数，代表两种操作之一（ i 为基于的历史版本号）：

    1.对于操作1，格式为 vi 1 loci valuei，即为在版本 vi 的基础上，将 aloci 修改为 valuei。

    2.对于操作2，格式为 vi 2 loci，即访问版本 vi 中的 aloci 的值。

(1 <= N，M <= 105，1 <= loci <= N，1 <= vi <= i，- 109 <= ai，valuei <= 109)

输出
输出包含若干行，依次为每个操作2的结果。
样例输入
5 10
59 46 14 87 41
0 2 1
0 1 1 14
0 1 1 57
0 1 1 88
4 2 4
0 2 5
0 2 4
4 2 1
2 2 2
1 1 5 91
样例输出
59
87
41
87
88
46

提示
询问生成的版本是指你访问的那个版本的复制
样例说明：
一共11个版本，编号从0-10，依次为：

• 0 : 59 46 14 87 41
• 1 : 59 46 14 87 41
• 2 : 14 46 14 87 41
• 3 : 57 46 14 87 41
• 4 : 88 46 14 87 41
• 5 : 88 46 14 87 41
• 6 : 59 46 14 87 41
• 7 : 59 46 14 87 41
• 8 : 88 46 14 87 41
• 9 : 14 46 14 87 41
• 10 : 59 46 14 87 91

主席树的思想

#include<bits/stdc++.h>
using namespace std;
const int maxn=2000000+5;

struct tree{
    int ls,rs,v; //ls：左儿子节点编号号  rs：右儿子节点编号   v：节点值 可持久化数组的节点值只在最底层赋值 也就是区间l=r时
}tree[maxn];
int num[maxn];   //初始数组
int total,edit[maxn];   //total节点编号  edit:版本数组  值为版本号也就是根节点的编号

int build(int l,int r){   //建线段树
   int now=++total;  
   if(l==r){
        tree[now].v=num[l];
        return now;
   }
   int mid=l+r>>1;
   tree[now].ls=build(l,mid);
   tree[now].rs=build(mid+1,r);
   return now;
}

int update(int ed,int l,int r,int p,int change){   //ed为版本号 l，r是区间 p为查询值  change为改变值
    int now=++total;  
    if(l==r){
        tree[now].v=change;
        return now;
    }
    tree[now].ls=tree[ed].ls;
    tree[now].rs=tree[ed].rs;
    int mid=l+r>>1;
    if(p<=mid)
        tree[now].ls=update(tree[ed].ls,l,mid,p,change);
    else
        tree[now].rs=update(tree[ed].rs,mid+1,r,p,change);
    return now;
}

int query(int ed,int l,int r,int p){   //查询  ed为版本号 l，r是区间 p为查询值
     if(l==r)
        return tree[ed].v;
      int mid=l+r>>1;
      if(p<=mid)
        return  query(tree[ed].ls,l,mid,p);
      else
        return  query(tree[ed].rs,mid+1,r,p);
}

int main(){
      int n,m;
      cin>>n>>m;
      for(int i=1;i<=n;i++)
          cin>>num[i];
      build(1,n);
      edit[0]=1;  //第一个版本的根节点是节点1
      for(int i=1;i<=m;i++){
          int vi,op,loci,value;
          cin>>vi>>op;
          if(op==1){
             cin>>loci>>value;
             edit[i]=update(edit[vi],1,n,loci,value);
          }else{
            cin>>loci;
            edit[i]=edit[vi];
             cout<<query(edit[vi],1,n,loci)<<endl;
          }
      }
}