这里推荐一篇讲解主席树的博客:点击这里
主席树主要解决的问题就是区间第K小数的问题。
主席树为什么叫主席树呢?
因为发明它的folite被我们叫做folite主席,所以就叫主席树。主席树也叫做可持久化线段树,因为其中的每一颗树都是一颗权值线段树,可持久化就是他可以保存的历史版本,因为线段树每次修改只会修改一个点或者一条链的信息。
- 以前缀和形式建立可持久化线段树
- 基于动态开点的储存形式
- 每次插入一个值新开 O ( l o g n ) O(logn) O(logn)个节点
- 空间复杂度 O ( n l o g n ) O(nlogn) O(nlogn)
- 单次操作时间复杂度 O ( l o g n ) O(logn) O(logn)
- 可以查询区间的值域信息
- 相对于线段树套平衡树的优势:代码简单速度快
- 劣势:离线结构难以修改:可以采用对询问分块的方式进行弥补。
当确定了元素个数n,或者key的范围[1,U],建出的线段树形态是唯一的。
对两棵key的上界相同的线段树进行参数相同的单点更新/区间询问时,所访问到的节点也是一致的。
所以当两棵线段树的下标和维护的范围大小都一样就可以说这两棵线段树可以合并。
对于每个位置都建一颗线段树,其实是一条链。
然后再按照顺序把线段树都合并起来。第i棵线段树维护的就是1~i区间的信息。
通过合并可以知道线段树满足可加性,那么得到的信息肯定可以相减。
对于每一次询问通过第r棵线段树和第(l-1)棵线段树作差,即可得到该区间的信息。
经过对左孩子作差,设左边的数字个数为c。
如果k<=c,那么左孩子查找第k大的。
如果k>c,那么在右孩子查找第(k-c)大的。
模板:
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mid (l+r)/2
using namespace std;
const int N = 200010;
int n, q, m, cnt = 0;
int a[N], b[N], T[N];
int sum[N<<5], L[N<<5], R[N<<5];
inline int build(int l, int r)
{
int rt = ++ cnt;
sum[rt] = 0;
if (l < r){
L[rt] = build(l, mid);
R[rt] = build(mid+1, r);
}
return rt;
}
inline int update(int pre, int l, int r, int x)
{
int rt = ++ cnt;
L[rt] = L[pre]; R[rt] = R[pre]; sum[rt] = sum[pre]+1;
if (l < r){
if (x <= mid) L[rt] = update(L[pre], l, mid, x);
else R[rt] = update(R[pre], mid+1, r, x);
}
return rt;
}
inline int query(int u, int v, int l, int r, int k)
{
if (l >= r) return l;
int x = sum[L[v]] - sum[L[u]];
if (x >= k) return query(L[u], L[v], l, mid, k);
else return query(R[u], R[v], mid+1, r, k-x);
}
int main()
{
scanf("%d%d", &n, &q);
for (int i = 1; i <= n; i ++){
scanf("%d", &a[i]);
b[i] = a[i];
}
sort(b+1, b+1+n);
m = unique(b+1, b+1+n)-b-1;
T[0] = build(1, m);
for (int i = 1; i <= n; i ++){
int t = lower_bound(b+1, b+1+m, a[i])-b;
T[i] = update(T[i-1], 1, m, t);
}
while (q --){
int x, y, z;
scanf("%d%d%d", &x, &y, &z);
int t = query(T[x-1], T[y], 1, m, z);
printf("%d\n", b[t]);
}
return 0;
}