主席树求区间第K小模板

主席树(President Tree)是一种用于解决区间查询和修改问题的数据结构,通常用于静态区间问题(即查询和修改操作在构建结构之后不再发生变化)。主席树可以高效地处理诸如区间和、区间最值等问题。

主席树的实现原理:

  1. 基本思想:主席树是一种基于分治思想的数据结构,它将原始序列按照每个位置的取值范围进行离散化,然后构建出一棵持久化线段树(Persistent Segment Tree)。

  2. 持久化线段树:持久化线段树是线段树的一种扩展,它能够记录每次修改后的线段树版本,而不是像传统线段树那样每次都覆盖更新。通过这种方式,我们可以在保留历史版本的基础上进行有效的区间查询。

  3. 构建过程:首先对原始序列进行排序和离散化,然后根据每个位置的取值范围构建出持久化线段树。在构建过程中,每棵线段树都会继承前一棵线段树的所有信息,并根据具体的修改操作进行更新。

  4. 查询操作:利用持久化线段树,可以高效地进行区间查询操作,例如区间求和、区间最值等。

优点和应用:

  1. 主席树适用于静态区间问题,即查询和修改操作在构建结构之后不再发生变化的场景。
  2. 它能够高效地处理区间查询和修改问题,例如区间和、区间最值等。

总的来说,主席树是一种用于解决静态区间查询和修改问题的数据结构,通过离散化和持久化线段树的方式,能够高效地处理各种区间操作。希望这些信息能帮助你理解主席树的基本原理和应用场景。

题目

给定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

Code

#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 小的元素。

你可能感兴趣的:(算法,c++,数据结构,主席树)