【模板】可持久化线段树 1（可持久化数组）

题目链接https://www.luogu.com.cn/problem/P3919

代码

#include<iostream>
#include<string>
#include<map>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<stack>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f
#define P pair<int,int>
#define ls p<<1
#define rs p<<1|1
using namespace std;
const int N=1e6+10;
int r[N],a[N],top;
struct node
{
	int l;
	int r;
	int v;
}t[N<<5];
int clone(int p)
{
	++top;
	t[top]=t[p];
	return top;
}
int build(int p,int l,int r)
{
	p=++top;
	if(l==r)
	{
		t[p].v=a[l];
		return top;
	}
	int mid=(l+r)>>1;
	t[p].l=build(t[p].l,l,mid);
	t[p].r=build(t[p].r,mid+1,r);
	return p;
}
int update(int p,int l,int r,int pos,int v)
{
	p=clone(p);
	if(l==r)
	{
		t[p].v=v;
	}
	else
	{
		int mid=(l+r)>>1;
		if(pos<=mid) t[p].l=update(t[p].l,l,mid,pos,v);
		else t[p].r=update(t[p].r,mid+1,r,pos,v);
	}
	return p;
}
int query(int p,int l,int r,int pos)
{
	if(l==r)
	{
		return t[p].v;
	}
	int mid=(l+r)>>1;
	if(pos<=mid) return query(t[p].l,l,mid,pos);
	else return query(t[p].r,mid+1,r,pos);
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	r[0]=build(0,1,n);
	for(int i=1;i<=m;i++)
	{
		int id,op,pos,v;
		scanf("%d%d%d",&id,&op,&pos);
		if(op==1)
		{
			scanf("%d",&v);
			r[i]=update(r[id],1,n,pos,v);
		}
		else
		{
			printf("%d\n",query(r[id],1,n,pos));
			r[i]=r[id];
		}
	}
    return 0;
}