【模板】可持久化线段树 1（主席树）

题目背景

这是个非常经典的主席树入门题——静态区间第K小
数据已经过加强，请使用主席树。同时请注意常数优化

题目描述

如题，给定N个正整数构成的序列，将对于指定的闭区间查询其区间内的第K小值。

输入输出格式

输入格式：
第一行包含两个正整数N、M，分别表示序列的长度和查询的个数。
第二行包含N个正整数，表示这个序列各项的数字。
接下来M行每行包含三个整数 l,r,k , 表示查询区间[l,r]内的第k小值。

输出格式：
输出包含k行，每行1个正整数，依次表示每一次查询的结果

输入输出样例

输入样例#1：
5 5
25957 6405 15770 26287 26465
2 2 1
3 4 1
4 5 1
1 2 2
4 4 1
输出样例#1：
6405
15770
26287
25957
26287
说明

数据范围：

对于20%的数据满足：1≤N,M≤10
对于50%的数据满足：1≤N,M≤10^3
对于80%的数据满足：1≤N,M≤10^5
对于100%的数据满足：1≤N,M≤2⋅10^5
对于数列中的所有数a_i ，均满足-10^9≤a_i≤10^9

样例数据说明：

N=5，数列长度为5，数列从第一项开始依次为[25957,6405,15770,26287,26465]
第一次查询为[2,2]区间内的第一小值，即为6405
第二次查询为[3,4]区间内的第一小值，即为15770
第三次查询为[4,5]区间内的第一小值，即为26287
第四次查询为[1,2]区间内的第二小值，即为25957
第五次查询为[4,4]区间内的第一小值，即为26287

卡了好久的主席树终于A了，延续喜好继续写指针

代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cstdlib>
using namespace std;
const int N=200010;
int n,q,m,cnt,a[N],b[N];

struct node
{
    node* ch[2];int s;
};
node* T[N];

void build(node* &o,int l,int r)
{
    if(l > r) return ;
    o=new node();
    if(l == r) return ;
    int mid=(l+r)>>1;
    build(o->ch[0],l,mid);
    build(o->ch[1],mid+1,r);
}

void updata(node* pre,node* &o,int l,int r,int x)
{
    if(l > r) return ;
    o=new node();o->ch[0]=pre->ch[0];
    o->ch[1]=pre->ch[1];o->s=pre->s+1;
    if(l == r) return ;
    int mid=(l+r)>>1;
    if(x <= mid) updata(pre->ch[0],o->ch[0],l,mid,x);
    else updata(pre->ch[1],o->ch[1],mid+1,r,x);
}

int query(node* u,node* v,int l,int r,int k)
{
    if(l >= r) return l;if(u == NULL || v == NULL) return 0;
    int s1=(u->ch[0] == NULL ? 0 : u->ch[0]->s),s2=(v->ch[0] == NULL ? 0 : v->ch[0]->s);
    int x=s2-s1,mid=(l+r)>>1;
    if(x >= k) return query(u->ch[0],v->ch[0],l,mid,k);
    return query(u->ch[1],v->ch[1],mid+1,r,k-x);
}

int read()
{
    int out=0,f=1;char c=getchar();
    while(c > '9' || c < '0') 
        {if(c == '-') f=-1;c=getchar();}
    while(c <= '9' && c >= '0') 
        {out=(out<<1)+(out<<3)+c-'0';c=getchar();}
    return out*f;
}

void solve()
{
    n=read(),q=read();
    for(int i=1;i<=n;i++) b[i]=a[i]=read();
    sort(b+1,b+n+1);
    m=unique(b+1,b+1+n)-b - 1;
    build(T[0],1,m);
    for(int i=1;i<=n;i++)
    {
        int t=lower_bound(b+1,b+m+1,a[i])-b;
        updata(T[i-1],T[i],1,m,t);
    }
    while(q--)
    {
        int x=read(),y=read(),z=read();
        int t=query(T[x-1],T[y],1,m,z);
        printf("%d\n",b[t]);
    }
}

int main()
{
    solve();
    return 0;
}