可持久化线段树/主席树 学习笔记

medal_dreams2026-04-18 20:19

定义

我们引用一下 OIWIKI 的定义：

可持久化数据结构 (Persistent data structure) 总是可以保留每一个历史版本，并且支持操作的不可变特性 (immutable)． --OIWIKI

故可持久化线段树就是可以存储多个历史版本，并且进行一些操作的线段树。

用途

可持久化线段树是很多可持久化的起点，它允许查询，修改某一版本的某一值，或者查询静态区间第 \(k\) 小。

值得一提的是，允许访问或修改所有版本的可持久化叫做完全可持久化。

详见：可持久化数据结构简介-OIWIKI。

基础实现

对于多个版本，我们很明显可以对于每一个版本建一个线段树，但是这样，我们的空间复杂度会爆炸！

故我们考虑让前面的版本对当前版本有一定贡献，考虑对于每一个版本，有以下两种情况：

  1. 如果没有修改的，我们就让指针指向上一个版本的这个未修改节点。
  2. 如果有修改的，我们就对其新建一个节点。
    如图：

其实现依赖于用结构体维护 \(ls, rs\)，并且对于一个新节点，我们先让其等于原节点，并且给其一个新的编号，再依次更新下面的节点。

代码如下：

cpp 复制代码 
struct node{
	int val, ls, rs;
}nd[N << 7];
#define ls(k) nd[k].ls
#define rs(k) nd[k].rs
int newnode(int p){nd[++ tot] = nd[p];return tot;}

修改操作

可见，因为我们将一个节点赋了一个新的编号，所以我们需要让这个节点变为这个编号，其实现方法即为在 update 操作中将引入 \(p\) 编号的节点加上址传递 ，即加上取地址符 &。其他和普通线段树没有什么很大的差异。

代码如下：

cpp 复制代码 
void update(int &p, int l, int r, int x, int val)
{
	p = newnode(p);//变为新编号
	if(l == r){nd[p].val = val;return ;}//更新值
	int mid = l + r >> 1;
	if(x <= mid) update(ls(p), l, mid, x, val);
	else update(rs(p), mid + 1, r, x, val);
	//pushup(p);
     //pushup 操作看题，可有可无
}

查询操作

因为所有版本我们已经处理好了，直接查询即可。

代码如下：

cpp 复制代码 
ll query(int p, int l, int r, int x)
{
	if(l == r) return nd[p].val;
	int mid = l + r >> 1;
	if(x <= mid) return query(ls(p), l, mid, x);
	else return query(rs(p), mid + 1, r, x);
}

基础实现例题

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

简要题意：

维护一个长度为 \(n\) 的数组 \(a\)，进行 \(m\) 次以下操作：

  1. 形如 v 1 p c 表示把版本 \(v\) 的 \(a_p\) 改为 \(c\)。
  2. 形如 v 2 p 输出版本 \(v\) 的 \(a_p\)。

对于每个操作，都会生成一个对应的版本。

其中 \(1 \le n,m \le 10 ^ 6\)，\(1 \le p \le n\)，\(-10^9 \le a_i,c \le 10^9\)。

如果当前是第 \(x\) 次操作，则 \(0 \le v < x\)。

版本 \(0\) 为原始数组。

思路

按照上面代码和思想，模拟即可。

代码：

cpp 复制代码 
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr int N = 1e6 + 66;
int root[N], tot;
struct Seg_Ment_Tree{
	struct node{
		int val, ls, rs;
	}nd[N << 7];
    #define ls(k) nd[k].ls
    #define rs(k) nd[k].rs
	int newnode(int p){nd[++ tot] = nd[p];return tot;}
	void update(int &p, int l, int r, int x, int val)
	{
		p = newnode(p);
		if(l == r){nd[p].val = val;return ;}
		int mid = l + r >> 1;
		if(x <= mid) update(ls(p), l, mid, x, val);
		else update(rs(p), mid + 1, r, x, val);
	}
	ll query(int p, int l, int r, int x)
	{
		if(l == r) return nd[p].val;
		int mid = l + r >> 1;
		if(x <= mid) return query(ls(p), l, mid, x);
		else return query(rs(p), mid + 1, r, x);
	}
}seg;
int n, m, a;
int main()
{
	ios::sync_with_stdio(false);cin.tie(0), cout.tie(0);
	cin >> n >> m;
	for(int i = 1;i <= n;i ++) cin >> a, seg.update(root[0], 1, n, i, a);//初始化初始版本（即原数组）
	for(int i = 1;i <= m;i ++)
	{
		int od, v, p, c;
		cin >> v >> od >> p;
		root[i] = root[v];//当前版本为给定版本
		if(od == 1)
		{
			cin >> c;
			seg.update(root[i], 1, n, p, c);//修改
		}
		else cout << seg.query(root[i], 1, n, p) << '\n';
	}
	return 0;
}

2.P1383 高级打字机

简要题意：

开始有一空串，要进行 \(n\) 次以下操作：

  1. 形如 T x，表示在后面加入一字符 \(x\)。
  2. 形如 U x，表示撤销 \(x\) 次除操作三的操作。
  3. 形如 Q x 表示查询当前字符串的第 \(x\) 个字符。

\(n \le 10^5\)。

思路

板子，按照模板 1 查询即可，注意输入的字符还是数字，不然会 WA，还有注意分辨一下操作二和操作一都属于修改操作，所以需要新建一个版本，具体看代码。

cpp 复制代码 
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr int N = 1e5 + 66;
int root[N], tot;
struct Seg_Ment_Tree{
	struct node{
		int val, ls, rs, su;
	}nd[N << 7];
#define ls(k) nd[k].ls
#define rs(k) nd[k].rs
	int newnode(int p){nd[++ tot] = nd[p];return tot;}
	void pushup(int p){nd[p].su = nd[ls(p)].su + nd[rs(p)].su;}//维护字符个数
	void update(int &p, int l, int r, int x, int val)
	{
		p = newnode(p);
		if(l == r){nd[p].val = val;nd[p].su = 1;return ;}
		int mid = l + r >> 1;
		if(x <= mid) update(ls(p), l, mid, x, val);
		else update(rs(p), mid + 1, r, x, val);
		pushup(p);
	}
	ll query(int p, int l, int r, int x)
	{
		if(l == r) return nd[p].val;
		int mid = l + r >> 1;
		if(x <= mid) return query(ls(p), l, mid, x);
		else return query(rs(p), mid + 1, r, x);
	}
}seg;
int n, m, a, cnt;
int main()
{
	ios::sync_with_stdio(false);cin.tie(0), cout.tie(0);
	cin >> n;
	for(int i = 1;i <= n;i ++)
	{
		char c;cin >> c;
		if(c == 'T')
		{
			char s;cin >> s;
			cnt ++;//新建版本
			root[cnt] = root[cnt - 1];//等于上一个版本
			seg.update(root[cnt], 1, 1e5, seg.nd[root[cnt]].su + 1, s);//在上一个版本的后面加入一个字符
		}
		else if(c == 'U')
		{
			int s;cin >> s;
			cnt ++;//新建版本
			root[cnt] = root[cnt - 1 - s];//等于 cnt - s 的，但是我们 ++ 了，所以让 cnt - 1
		}
		else 
		{
			int s;cin >> s;
			cout << char(seg.query(root[cnt], 1, 1e5, s)) << '\n';//直接查询即可
		}
	}
	return 0;
}

进阶一点的实现

现在我们讨论静态区间第 \(k\) 小。

先引入一下例题吧：

P3834 【模板】可持久化线段树 2

简要题意：

给定一个长度为 \(n\) 的序列 \(a\)，询问 \(m\) 次 \(a\) 中 \([l,r]\) 区间第 \(k\) 小的数的值。
\(1 \le n,m \le 2 \times 10 ^ 5\)，\(0 \le a_i \le 10^9\)，\(1 \le l, r \le n\)，\(1 \le k \le r - l + 1\)。

可见，我们很难在原序列中，用原本思想直接在 \(i\) 修改，因为我们第 \(k\) 小是关系大小的。

那大小我们怎么处理呢？考虑使用权值线段树来处理，这样我们就可以处理大小关系。

那太大了怎么办？可持久化线段树是自带动态开点性质的，我们只需关注题目的空间大小是不是过小了（比如 \(125MB\)），如果太小了，就离散化一下即可。

修改

对于每一个位置，我们考虑建立一个新版本，并在新版本所维护的权值线段树上插入它的值。

所以修改代码其实是没什么区别的，但是 \(a_i\) 可能重复，所以不能直接赋值。

考虑我们还需要维护什么，以及我们需不需要 pushup。显然的是我们需要查询前面区间的数的个数，所以我们需要 pushup 来让父节点维护区间。

代码如下：

cpp 复制代码 
void pushup(int p){nd[p].val = nd[ls(p)].val + nd[rs(p)].val;}
void update(int &p, int l, int r, int x, int val)
{
	p = newnode(p);
	if(l == r){nd[p].val += val;return ;}
	int mid = l + r >> 1;
	if(x <= mid) update(ls(p), l, mid, x, val);
	else update(rs(p), mid + 1, r, x, val);
	pushup(p);
}

查询

但是我们怎么查询的？我们考虑，对于 \([l,r]\) 我们现在已经有了对应的版本，那一个 \([1,l]\)，\([1,r]\) 内有的数的个数的差不就是 \([l,r]\) 中数的个数减去一吗？考虑到这个也可以推广到其他区间，所以我们在查询时传入两个节点，每一次判断前面的区间的数是否够 \(k\)，如果不够就在后面的区间里，但是此时前面是有贡献的，所以注意此时查询的 \(k\) 就要减去前面的数的个数了。

注意传入的节点是 \([l-1,r]\) 即可。

对了，此时两个节点要同时跳左右儿子哦。

代码如下：

cpp 复制代码 
ll query(int u, int v, int l, int r, int x)
{
	if(l == r) return l;//因为我们建的是权值线段树，此时 l 就是我们对应的值
	int k = nd[ls(v)].val - nd[ls(u)].val;
	int mid = l + r >> 1;
	if(k >= x) return query(ls(u), ls(v), l, mid, x);
	else return query(rs(u), rs(v), mid + 1, r, x - k);
}

全部代码

cpp 复制代码 
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr int N = 1e6 + 66;
int root[N], tot;
struct Seg_Ment_Tree{
	struct node{
		int val, ls, rs;
	}nd[N << 7];
#define ls(k) nd[k].ls
#define rs(k) nd[k].rs
	int newnode(int p){nd[++ tot] = nd[p];return tot;}
	void pushup(int p){nd[p].val = nd[ls(p)].val + nd[rs(p)].val;}
	void update(int &p, int l, int r, int x, int val)
	{
		p = newnode(p);
		if(l == r){nd[p].val += val;return ;}
		int mid = l + r >> 1;
		if(x <= mid) update(ls(p), l, mid, x, val);
		else update(rs(p), mid + 1, r, x, val);
		pushup(p);
	}
	ll query(int u, int v, int l, int r, int x)
	{
		if(l == r) return l;
		int k = nd[ls(v)].val - nd[ls(u)].val;
		int mid = l + r >> 1;
		if(k >= x) return query(ls(u), ls(v), l, mid, x);
		else return query(rs(u), rs(v), mid + 1, r, x - k);
	}
}seg;
int n, m, a;
int main()
{
	ios::sync_with_stdio(false);cin.tie(0), cout.tie(0);
	cin >> n >> m;
	for(int i = 1;i <= n;i ++)
	{
		cin >> a;
		root[i] = root[i - 1];
		seg.update(root[i], 0, 1e9, a, 1);
	}
	for(int i = 1;i <= m;i ++)
	{
		int l, r, k;cin >> l >> r >> k;
		cout << seg.query(root[l - 1], root[r], 0, 1e9, k) << '\n';
	}
	return 0;
}

例题

1.P1533 可怜的狗狗

简要题意是和上面一样的，不重新打一遍了。

数据范围：\(1\le n \le 3\times 10^5 ,1\le m \le5\times10^4,0\le a_i<2^{31}\)，且 \(a_i\) 互不相同。

其实这个就是模板二，注意空间限制比较小，离散化一下就可以了。

cpp 复制代码 
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr int N = 3e5 + 66;
int root[N], tot, inf = INT_MAX;
struct Seg_Ment_Tree{
	struct node{
		int val, ls, rs;
	}nd[N << 5];
#define ls(k) nd[k].ls
#define rs(k) nd[k].rs
	int newnode(int p){nd[++ tot] = nd[p];return tot;}
	void pushup(int p){nd[p].val = nd[ls(p)].val + nd[rs(p)].val;}
	void update(int &p, int l, int r, int x, int val)
	{
		p = newnode(p);
		if(l == r){nd[p].val += val;return ;}
		int mid = l + r >> 1;
		if(x <= mid) update(ls(p), l, mid, x, val);
		else update(rs(p), mid + 1, r, x, val);
		pushup(p);
	}
	int query(int u, int v, int l, int r, int x)
	{
		if(l == r) return l;
		int k = nd[ls(v)].val - nd[ls(u)].val;
		int mid = l + r >> 1;
		if(k >= x) return query(ls(u), ls(v), l, mid, x);
		else return query(rs(u), rs(v), mid + 1, r, x - k);
	}
}seg;
int n, m, a[N], b[N];
int main()
{
	ios::sync_with_stdio(false);cin.tie(0), cout.tie(0);
	cin >> n >> m;
	for(int i = 1;i <= n;i ++)
	{
		cin >> a[i];b[i] = a[i];
	}
	sort(b + 1, b + 1 + n);
	for(int i = 1;i <= n;i ++) a[i] = lower_bound(b + 1, b + 1 + n, a[i]) - b;
	for(int i = 1;i <= n;i ++) 
	{
		root[i] = root[i - 1];
		seg.update(root[i], 1, n, a[i], 1);
	}
	for(int i = 1;i <= m;i ++)
	{
		int l, r, k;cin >> l >> r >> k;
		cout << b[seg.query(root[l - 1], root[r], 1, n, k)] << '\n';
	}
	return 0;
}

2.P3168 [CQOI2015] 任务查询系统

简要题意：

有 \(n\) 个任务，其统治区间为 \([l_i,r_i]\)，权值为 \(p_i\)。

询问 \(m\) 次，强制在线，询问对于 \(x_i\)，权值最小的 \(k_i\) 的任务的权值和。

\(1\le m,n \le 10 ^ 5\)，\(1\leq l_i\leq r_i\leq n\)，\(1\le p_i \le 10^7\)，\(x_i\) 为 \(1\) 到 \(n\) 的一个排列。

这不是逗我呢吗？《

主席树不是只能单点插入吗，怎么还要区间插入？？？

等等，你别急，主席树虽然只能单点插入，但是它的版本传递有类似于前缀和的作用，所以我们对于 \(l_i\) 和 \(r_i + 1\) 进行权值线段树，但是差分就可以了。注意一个位置的数可以有许多，所以注意使用 vector 记录每个数。

查询时注意是权值和，所以我们要把左区间的贡献也算上。

代码:

cpp 复制代码 
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr int N = 1e5 + 66;
struct Persistent_Segment_Tree{
	struct node{
		ll val, su;
		int rs, ls;
	}nd[N << 7];
	int tot;
#define ls(k) nd[k].ls
#define rs(k) nd[k].rs
#define val(k) nd[k].val
#define su(k) nd[k].su
	void pushup(int p){val(p) = val(ls(p)) + val(rs(p));su(p) = su(ls(p)) + su(rs(p));}
	int newnode(int p){nd[++ tot] = nd[p];return tot;}
	void update(int &p, int l, int r, ll x, ll val)
	{
		p = newnode(p);
		if(l == r){val(p) += val;su(p) += x * val;return ;}
		int mid = l + r >> 1;
		if(x <= mid) update(ls(p), l, mid, x, val);
		else update(rs(p), mid + 1, r, x, val);
		pushup(p);
	}
	ll query(int p, int l, int r, ll x)
	{
		if(l == r) return min(su(p), x * l);//注意可能有多个 l，比较一下取最小即可
		int mid = l + r >> 1;
		if(x <= val(ls(p))) return query(ls(p), l, mid, x);
		else return query(rs(p), mid + 1, r, x - val(ls(p))) + su(ls(p));
	}
}seg;
ll Abs(ll x){return x > 0 ? x : -x;}
ll n, m, mx;
int root[N];
vector<ll> v[N];
int main()
{
	ios::sync_with_stdio(false);cin.tie(0), cout.tie(0);
	cin >> m >> n;
	for(int i = 1;i <= m;i ++)
	{
		ll x, y, z;cin >> x >> y >> z;
		v[x].push_back(z), v[y + 1].push_back(-z);
		mx = max(mx, z);
	}
	for(int i = 1;i <= n;i ++)
	{
		root[i] = root[i - 1];
		for(auto p : v[i]) seg.update(root[i], 1, mx, Abs(p), (p > 0 ? 1ll : -1ll));
	}
	ll lst = 1;
	for(int i = 1;i <= n;i ++)
	{
		ll x, a, b, c;cin >> x >> a >> b >> c;
		a = (lst * a + b) % c + 1;
		lst = seg.query(root[x], 1, mx, a);
		cout << lst << '\n';
	}
	return 0;
}
相关推荐
大佬，救命！！！1 天前
etp中未运行用例顺序的定位及补齐脚本自动化生成
python·学习笔记·excel·自动化脚本·用例整理清洗
其实防守也摸鱼3 天前
ctfshow--VIP题目限免（包含原理和知识拓展）前10个
网络·算法·安全·学习笔记·ctf·泄露·web类型
四谎真好看3 天前
Redis学习笔记（高级篇3）
redis·笔记·学习·学习笔记
其实防守也摸鱼4 天前
Web漏洞全景解析：从原理溯源到实战攻防的进阶指南
网络·web安全·网络安全·学习笔记·web类型漏洞
551只玄猫6 天前
新编大学德语1第三版笔记 第7课Kaufen und Schenken
学习笔记·德语·外语·德语a1·德语入门·自学外语·新编大学德语
四谎真好看8 天前
Redis学习笔记（高级篇2）
redis·笔记·学习·学习笔记
2401_8920709811 天前
算法与数据结构精讲：最大子段和（暴力 / 优化 / 分治）+ 线段树从入门到实战
c++·算法·线段树·最大子段和
圣光SG11 天前
项目分析与程序设计 学习笔记
笔记·学习·学习笔记·程序设计·项目分析
四谎真好看14 天前
Redis学习笔记（实战篇5 + 高级篇）
redis·笔记·学习·学习笔记
热门推荐
012026年4月技术前沿：AI大模型爆发、智能体革命与量子安全新纪元02GitHub 镜像站点032026 年 AI 编程助手全面对比评测：Cursor vs Copilot vs Claude Code vs GitHub Copilot Free04AI Weekly | 2026年4月第二周 · GitHub热门项目与AI发展趋势深度解析05GPT-6发布日深度解析-Symphony架构200万Token实战06零成本！Ollama本地部署国产大模型全指南（支持Kimi-K2.5/GLM-5/Qwen，新手秒上手）07一周AI热点速览（2026.03.31-04.06）：GPT-6曝光、谷歌开源Gemma 4、资本狂飙与模型军备竞赛08从零部署 Hermes Agent：一只"会成长的 AI 马"保姆级安装教程09从限购到畅通：GLM-5.1 Coding Plan接入攻略10基于 Docker 部署 Hermes Agent 并接入飞书机器人的完整指南