可持久化线段树
它可以记录线段树历史状态,对历史状态进行查询。
又名主席树。
因为线段树是从叶节点往上修改,而这个过程中只有修改一些节点,所以,别的节点可以不动。
可持久化线段树就是只对修改过的节点进行复制。别的节点保留不动。这样空间就不会炸。
如果复制一整棵树的话,空间复杂度是 O ( n ∗ m ) O(n*m) O(n∗m)。
图解:
图是我自己画的,可能丑了点。
红色表示原来的树的部分,黄色是第一次修改,蓝色是第二次修改。可以发现,蓝色是在黄色的基础上进行修改的。
这样子的好处就是空间的优化。
其余操作与线段树一样。
我调了好几次才过,因为,第一次,存树的数组开了4倍。RE了几个点。第二次,开了10倍,RE了两个点。第三次,愤怒地开了100倍,同时也把别的数组开了10倍,当然,这没必要。终于AC了。
CODE by lmrttx :
#include<bits/stdc++.h>
#define maxn 1000000
using namespace std;
struct node{
int l,r,val;
}tree[maxn*100];
int n,m,a[maxn*10],root[maxn*10],top;
int clean_(int id){
top++;
tree[top]=tree[id];
return top;
}
int build(int id,int l,int r){
top++;
id=top;
if(l==r){
tree[id].val=a[l];
return top;
}
int mid=(l+r)>>1;
tree[id].l=build(tree[id].l,l,mid);
tree[id].r=build(tree[id].r,mid+1,r);
return id;
}
int update(int id,int l,int r,int x,int val){
id=clean_(id);
if(l==r)tree[id].val=val;
else {
int mid=(l+r)>>1;
if(x<=mid)
tree[id].l=update(tree[id].l,l,mid,x,val);
else tree[id].r=update(tree[id].r,mid+1,r,x,val);
}
return id;
}
int query(int id,int l,int r,int x){
if(l==r)return tree[id].val;
else {
int mid=(l+r)>>1;
if(x<=mid)
return query(tree[id].l,l,mid,x);
else return query(tree[id].r,mid+1,r,x);
}
}
int main(){
scanf("%d%d",&n,&m);
for(register int i=1;i<=n;i++)scanf("%d",&a[i]);
root[0]=build(0,1,n);
for(register int i=1,root1,op,x;i<=m;i++){
scanf("%d%d%d",&root1,&op,&x);
switch (op){
case 1:
int y;
scanf("%d",&y);
root[i]=update(root[root1],1,n,x,y);
break;
case 2:
printf("%d\n",query(root[root1],1,n,x));
root[i]=root[root1];
break;
}
}
return 0;
}
就这样吧。谢谢阅读。