主席树(President Tree)是一种用于解决区间查询和修改问题的数据结构,通常用于静态区间问题(即查询和修改操作在构建结构之后不再发生变化)。主席树可以高效地处理诸如区间和、区间最值等问题。
基本思想:主席树是一种基于分治思想的数据结构,它将原始序列按照每个位置的取值范围进行离散化,然后构建出一棵持久化线段树(Persistent Segment Tree)。
持久化线段树:持久化线段树是线段树的一种扩展,它能够记录每次修改后的线段树版本,而不是像传统线段树那样每次都覆盖更新。通过这种方式,我们可以在保留历史版本的基础上进行有效的区间查询。
构建过程:首先对原始序列进行排序和离散化,然后根据每个位置的取值范围构建出持久化线段树。在构建过程中,每棵线段树都会继承前一棵线段树的所有信息,并根据具体的修改操作进行更新。
查询操作:利用持久化线段树,可以高效地进行区间查询操作,例如区间求和、区间最值等。
总的来说,主席树是一种用于解决静态区间查询和修改问题的数据结构,通过离散化和持久化线段树的方式,能够高效地处理各种区间操作。希望这些信息能帮助你理解主席树的基本原理和应用场景。
给定n
个数,q
次询问,每次求区间 [ l , r ]
中从小到大排序后第 k
个数。
1 <= n <= 100 000, 1 <= q <= 5 000,1 <=l <=r<= n, 1 <= k <= j - i + 1
Input
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3
Output
5
6
3
#include
#include
#include
#define midd (l+r)/2
using namespace std;
typedef long long ll;
const int N = 200010;
int n, q, sz, cnt = 0;
int a[N], b[N], T[N];//a为原数组,b为排序离散化数组,T[i]为(值)[1,i]的根节点
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, midd);
R[rt] = build(midd+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 <= midd) L[rt] = update(L[pre], l, midd, x);
else R[rt] = update(R[pre], midd+1, r, x);
}
return rt;
}
inline int query(int u, int v, int l, int r, int k)//返回下标,即<=k个数
{
if (l >= r) return l;
int x = sum[L[v]] - sum[L[u]];
if (x >= k) return query(L[u], L[v], l, midd, k);
else return query(R[u], R[v], midd+1, r, k-x);
}
int main()
{
while(scanf("%d%d", &n, &q)!=EOF){
cnt=0;
for (int i = 1; i <= n; i ++){
scanf("%d", &a[i]);
b[i] = a[i];
}
sort(b+1, b+1+n);
sz = unique(b+1, b+1+n)-b-1;
T[0] = build(1, sz);
for (int i = 1; i <= n; i ++){
int t = lower_bound(b+1, b+1+sz, a[i])-b;
T[i] = update(T[i-1], 1, sz, t);
}
while (q --){
int l, r, k;
scanf("%d%d%d", &l, &r ,&k);
// k= r-l+1-k+1;//求第k大时改一下就行
int idx=query(T[l-1], T[r], 1, sz, k);
int ans=b[idx];
printf("%d\n",ans);
}
}
return 0;
}
通过构建持久化线段树,实现了区间查询和修改操作。可以通过调用 update 函数更新结点信息,通过调用 query 函数查询区间内第 k 小的元素。