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

(Updated: )
OI
,
views
Word count: 612 (~3 mins to read) Last updated:

原题地址:https://www.luogu.org/problemnew/show/P3919

题目简述

维护一个长度为N的数组，支持如下几种操作：

  1. 在某个历史版本上修改某一个位置上的值
  2. 访问某个历史版本上的某一位置的值
    此外，每进行一次操作（对于操作2，即为生成一个完全一样的版本，不作任何改动），就会生成一个新的版本。版本编号即为当前操作的编号（从1开始编号，版本0表示初始状态数组）

思路

首先感谢来自@zcysky的模板。写的非常漂亮，封装也很精致。
这题是裸题，于是直接上模板就行了。注意此题输入数据大，需要读入优化。
关于可持久化线段树的介绍与总结，之后把坑填上。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
#include<bits/stdc++.h>
const int N=1000005;
using namespace std;
int a[N],n,q,rt[N*20];//空间复杂度O(mlogn) 
#define MAXB 50000000
char buf[MAXB],*cp=buf;
inline int read()
{
    int f=1,x=0;
    while(*cp<'0'||*cp>'9') {
        if(*cp=='-')
            f=-1;
        cp++;
    }
    while(*cp>='0'&&*cp<='9') {
        x=x*10+*cp-'0'; 
        cp++;
    }
    return f*x;
}
struct Persistable_Segment_Tree {
    int lc[N*20],rc[N*20],val[N*20],cnt;
    inline void build(int &o,int l,int r){
        o=++cnt;
        if(l==r) {
            val[o]=a[l];
            return;
        }
        int mid=(l+r)>>1;
        build(lc[o],l,mid);
        build(rc[o],mid+1,r);
    }
    inline void ins(int &o,int pre,int l,int r,int q,int v){
        o=++cnt;
        lc[o]=lc[pre];
        rc[o]=rc[pre];
        val[o]=val[pre];
        if(l==r) {
            val[o]=v;
            return;
        }
        int mid=(l+r)>>1;
        if(q<=mid)
            ins(lc[o],lc[pre],l,mid,q,v);
        else 
            ins(rc[o],rc[pre],mid+1,r,q,v);
    }
    inline int query(int o,int l,int r,int q){//类似二分的查询 
        if(l==r)
            return val[o];
        int mid=(l+r)>>1;
        if(q<=mid)
            return query(lc[o],l,mid,q);
        else 
            return query(rc[o],mid+1,r,q);
    }
}T;
int main()
{
    fread(buf,1,MAXB,stdin);
    n=read();
    int m=read();
    for(int i=1;i<=n;i++)
        a[i]=read();
    T.build(rt[0],1,n);
    for(int i=1;i<=m;i++){
        int pre=read(),opt=read(),x=read();
        if(opt==1) {//操作1:在版本pre的基础上将第x个数修改为v 
            int v=read();
            T.ins(rt[i],rt[pre],1,n,x,v);
        }
        if(opt==2) {//操作2:访问版本pre中第x个数的值 
            printf("%d\n",T.query(rt[pre],1,n,x));
            rt[i]=rt[pre];
        }
    }
}