主席树模板

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
#define inf 0x3f3f3f3f
const int maxn=1e5+9;
int cnt=0;
struct node{
    int val,id;
}a[maxn];
struct tree{
    int L,R,sum;
}tr[maxn*20];
bool cmp(node a,node b){
    return a.val<b.val;
}
int Rank[maxn];//离散化后的第i个点是第几小
int root[maxn];//前缀树的根节点的编号
void update(int num,int &rt,int l,int r){
    tr[++cnt]=tr[rt];//copy前一棵树
    rt=cnt;
    tr[rt].sum++;
    if(l==r)return ;
    int mid=(l+r)>>1;
    if(num<=mid)update(num,tr[rt].L,l,mid);
    else update(num,tr[rt].R,mid+1,r);
}
int query(int i,int j,int k,int l,int r){
    int d=tr[tr[j].L].sum-tr[tr[i].L].sum;
    if(l==r)return l;
    int mid=(l+r)>>1;
    if(k<=d)return query(tr[i].L,tr[j].L,k,l,mid);
    else return query(tr[i].R,tr[j].R,k-d,mid+1,r);
}
int main(){
    int i,j,k,n,m;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++){
        scanf("%d",&a[i].val);
        a[i].id=i;
    }
    sort(a+1,a+n+1,cmp);
    for(i=1;i<=n;i++){
        Rank[a[i].id]=i;
    }
    for(i=1;i<=n;i++){
        root[i]=root[i-1];
        update(Rank[i],root[i],1,n);
    }
    for(i=0;i<m;i++){
        int l,r;
        scanf("%d%d%d",&l,&r,&k);
        int pos=query(root[l-1],root[r],k,1,n);
        printf("%d\n",a[pos].val);
    }
}