前言
主席树的全称是可持久化权值线段树,关键词有两个,一个是可持久化,一个是权值线段树
主席树与普通线段树有不少区别,学主席树之前最好学一下权值线段树,
理解其蕴含的前缀存储思想
可以把权值线段树理解为一个桶,每个叶子节点leaf[k]存的是a[1, …, n]中,k出现的次数
因此在构建主席树之前应当将数据离散化处理一下,这里由dispersed()和find()完成
主席树与普通权值线段树的区别主要在于增加了可持久化功能,
具体表现为update()函数的动态开点功能(开出历史存档),
以及query对不同版本的树根查询(借鉴了权值线段树的前缀存储思想)
一、 例题 p3834
二、 思路及代码
1. 思路
很经典的主席树入门,直接套模板即可
2. 代码
代码如下 :
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn = 2e5 + 7;
int root[maxn], id, len = 1;
int a[maxn], b[maxn];
int n, m;
struct ptree {
int l, r;
int cnt; // 权重线段树
} t[maxn << 5];
int dispersed() {
for (int i = 1; i <= n; i++) b[i] = a[i];
sort(b + 1, b + n + 1);
int len = unique(b + 1, b + n + 1) - b - 1;
// for (int i = 1; i <= n; i++)
// a[i] = lower_bound(b + 1, b + len + 1, a[i]) - b;
return len;
}
int find(int x) {
int l = 0, r = len;
while (l < r) {
int mid = l + r >> 1;
if (b[mid] >= x)
r = mid;
else
l = mid + 1;
}
return l;
}
int build(int l, int r) {
int cur = ++id;
if (l == r) return cur;
int mid = l + r >> 1;
t[cur].l = build(l, mid);
t[cur].r = build(mid + 1, r);
return cur;
}
int update(int pre, int x, int l, int r) {
int cur = ++id;
int mid = l + r >> 1;
t[cur] = t[pre], t[cur].cnt++; // 权值++
if (l < r)
if (x <= mid) // 子树权值更新
t[cur].l = update(t[pre].l, x, l, mid);
else
t[cur].r = update(t[pre].r, x, mid + 1, r);
return cur;
}
int query(int lt, int rt, int l, int r, int k) {
if (l == r) return l;
int tmp = t[t[rt].l].cnt - t[t[lt].l].cnt; // 左子树的权重
int mid = l + r >> 1; // 即1 - tmp排名的节点均在左子树
if (k <= tmp) return query(t[lt].l, t[rt].l, l, mid, k);
return query(t[lt].r, t[rt].r, mid + 1, r, k - tmp);
}
int main() {
// freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
len = dispersed();
root[0] = build(1, len); // 离线构建权重线段树
for (int i = 1; i <= n; i++)
root[i] = update(root[i - 1], find(a[i]), 1, len);
while (m--) {
int l, r, k;
scanf("%d%d%d", &l, &r, &k); // 在线查询第k小
printf("%d\n", b[query(root[l - 1], root[r], 1, len, k)]);
}
return 0;
}