LUOGU P3919 模板可持久化数组(主席树)

Posted 2021-01-20 sdfzsyq

传送门

解题思路

　　给每一时刻建一棵线段树维护当前时刻的值，然后修改的时候直接修改，查询的时候直接查，记住查询完后一定要复制。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>

using namespace std;
const int MAXN = 1000005;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch==‘-‘?0:1;ch=getchar();}
    while(isdigit(ch))  {x=(x<<1)+(x<<3)+ch-‘0‘;ch=getchar();}
    return f?x:-x;
}

int n,m,a[MAXN],rt[MAXN],cnt;
int val[MAXN*22],ls[MAXN*22],rs[MAXN*22];

int build(int l,int r){
    int now=++cnt,mid=(l+r)>>1;
    if(l==r) {
        val[now]=rd();
        return now;
    }
    ls[now]=build(l,mid);rs[now]=build(mid+1,r);
    return now;
}

int update(int pre,int l,int r,int x,int k){
    int now=++cnt,mid=(l+r)>>1;
    ls[now]=ls[pre];rs[now]=rs[pre];val[now]=val[pre];
    if(l==r) {val[now]=k;return now;}
    if(x<=mid) ls[now]=update(ls[pre],l,mid,x,k);
    else rs[now]=update(rs[pre],mid+1,r,x,k);
    return now; 
}

int query(int pre,int l,int r,int x){
    if(l==r) return val[pre];
    int mid=(l+r)>>1;
    if(x<=mid) return query(ls[pre],l,mid,x);
    else return query(rs[pre],mid+1,r,x);
}

int main(){
    n=rd(),m=rd();rt[0]=build(1,n);int pre,op,x,y;
    for(int i=1;i<=m;i++){
        pre=rd(),op=rd(),x=rd();
        if(op==1) y=rd(),rt[i]=update(rt[pre],1,n,x,y);
        else printf("%d
",query(rt[pre],1,n,x)),rt[i]=rt[pre];
    }   
    return 0;
}