可持久化数组 (luogu3919)

传送门

可持久化数组支持单点查询某个状态的值 , 单点修改某个状态的值

具体实现如下 :

建一棵主席树 (不同于权值线段树 , 是普通的线段树)

最后一层的节点的值 , 就是当前版本那个点的值

于是查询一个版本某个节点的值, 查对应子树对应的值就可以了

#include<bits/stdc++.h>
#define N 1000050
using namespace std;

struct Node{
	int ls,rs,val;
}t[N*20];

int n,m,a[N],cnt;
int rt[N];

int read(){
	int cnt=0,f=1; char ch=0;
	while(!isdigit(ch)){ch=getchar(); if(ch=='-') f=-1;}
	while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt*f;
}
void Build(int &x,int l,int r){
	x = ++cnt;
	if(l==r){ t[x].val = a[l]; return;}
	int mid = (l+r)>>1;
	Build(t[x].ls, l, mid);
	Build(t[x].rs, mid+1, r);
}
void Update(int &x, int last, int L, int R,int pos,int val){
	x = ++cnt;  t[x] = t[last];
	if(L==R){ t[x].val = val; return;}
	int mid = (L+R) >> 1;
	if(pos<=mid) Update(t[x].ls, t[last].ls, L, mid, pos, val);
	else Update(t[x].rs, t[last].rs, mid+1, R, pos, val);
}
int Quary(int x,int L,int R,int pos){
	if(L==R) return t[x].val;
	int mid = (L+R) >> 1;
	if(pos<=mid) return Quary(t[x].ls, L, mid, pos);
	else return Quary(t[x].rs, mid+1, R, pos);
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++) a[i] = read();
	Build(rt[0],1,n);
	for(int i=1;i<=m;i++){
		int v=read(), op=read(), x=read();
		if(op==1){
			int val = read();
			Update(rt[i], rt[v], 1, n, x, val);
			// 在版本v的基础上, 建立第i个版本 
		}
		if(op==2){
			int ans = Quary(rt[v], 1, n, x);
			printf("%d\n",ans);  rt[i] = rt[v];
		}
	}  return 0;
}