洛谷 P3919 【模板】可持久化数组（可持久化线段树/平衡树）

题目链接：https://www.luogu.org/problem/P3919

思路：主席树...........

#include<string.h>
#include<stdio.h>
#include<algorithm>
#define M 1000010
using namespace std;
struct node
{
    int l,r,w;
}T[20*M];
int tot,n,m;
int root[M],A[M];
void build(int &rt,int l,int r)
{
    rt=++tot;
    if(l==r)
    {
        T[rt].w=A[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(T[rt].l,l,mid);
    build(T[rt].r,mid+1,r);
    T[rt].w=T[T[rt].l].w+T[T[rt].r].w;
}
void update(int &rt,int old,int pos,int w,int l,int r)
{
    T[++tot]=T[old];
    rt=tot;
    if(l==r)
    {
        T[rt].w=w;
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) update(T[rt].l,T[old].l,pos,w,l,mid);
    else update(T[rt].r,T[old].r,pos,w,mid+1,r);
    T[rt].w=T[T[rt].l].w+T[T[rt].r].w;
}
int query(int rt,int pos,int l,int r)
{
    if(l==r) return T[rt].w;
    int mid=(l+r)>>1;
    if(pos<=mid) return query(T[rt].l,pos,l,mid);
    else return query(T[rt].r,pos,mid+1,r);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&A[i]);
    build(root[0],1,n);
    int v,opt,pos,w;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&v,&opt,&pos);
        if(opt==1)
        {
            scanf("%d",&w);
            update(root[i],root[v],pos,w,1,n);
        }
        else
        {
            root[i]=root[v];
            printf("%d\n",query(root[i],pos,1,n));
        }
    }
}